model select

app.py

@@ -15,13 +15,15 @@ st.set_page_config(
 
 st.header("Do It!!!! Ask the Question.....")
 
-question=st.text_input("Question:")
+col1,col2=st.columns([5,2])
 
+question=col1.text_input("Question:")
+model_name=st.selectbox("Model Name",options=["google/flan-t5-large","meta-llama/Llama-2-7b-chat-hf"])
 
 submit=st.button("Submit")
 
 if submit:
     with st.spinner("In progress..."):
-        response=get_answer(question)
+        response=get_answer(question,model_name)
     st.subheader("Response")
     st.write(f"{response}")