infernece api models

app.py

@@ -15,10 +15,10 @@ st.set_page_config(
 
 st.header("Do It!!!! Ask the Question.....")
 
-col1,col2=st.columns([5,2])
+col1,col2=st.columns([5,3])
 
 question=col1.text_input("Question:")
-model_name=st.selectbox("Model Name",options=["google/flan-t5-large","meta-llama/Llama-2-7b-chat-hf"])
+model_name=col2.selectbox("Model Name",options=["google/flan-t5-large","gpt2-medium","EleutherAI/gpt-neo-125m"])
 
 submit=st.button("Submit")