Update app.py

app.py

@@ -109,39 +109,22 @@ with st.sidebar:
             data_ingestion()  # Process PDF every time new file is uploaded
             st.success("Done")
     
-    genre = st.radio(
+    model_name_select = st.radio(
     "Please select LLM",
-    [":rainbow[mistralai/Mistral-7B-Instruct-v0.2]",":rainbow[google/gemma-7b-it]",":rainbow[google/gemma-1.1-7b-it]",]
-     #":rainbow[deepset/roberta-base-squad2]",":rainbow[consciousAI/question-answering-roberta-base-s-v2]",
-     #":rainbow[distilbert/distilbert-base-cased-distilled-squad]"]
-    #captions = ["Laugh out loud.", "Get the popcorn.", "Never stop learning.","Get the popcorn.", "Never stop learning."]
+    [":rainbow[mistralai/Mistral-7B-Instruct-v0.2]",":rainbow[google/gemma-1.1-7b-it]"]
+    
     )
 
-    if genre == ':rainbow[mistralai/Mistral-7B-Instruct-v0.2]':
+    if model_name_select == ':rainbow[mistralai/Mistral-7B-Instruct-v0.2]':
         st.write('You selected Mistral-7B-Instruct-v0.2.')
         model_name_query="mistralai/Mistral-7B-Instruct-v0.2"
         flag = True
-    elif genre == ':rainbow[google/gemma-7b-it]':
-        st.write('You selected google/gemma-7b-it.')
-        model_name_query="google/gemma-7b-it"
-        flag = True
-    elif genre == ':rainbow[google/gemma-1.1-7b-it]':
+    
+    elif model_name_select == ':rainbow[google/gemma-1.1-7b-it]':
         st.write('You selected HuggingFaceH4/zephyr-7b-gemma-v0.1')
         model_name_query="google/gemma-1.1-7b-it"
         flag = True
-    elif genre == ':rainbow[distilbert/distilbert-base-cased-distilled-squad]':
-        st.write('You selected distilbert/distilbert-base-cased-distilled-squad')
-        model_name_query="distilbert/distilbert-base-cased-distilled-squad"
-        flag = False
-    elif genre == ':rainbow[consciousAI/question-answering-roberta-base-s-v2]':
-        st.write('You selected consciousAI/question-answering-roberta-base-s-v2')
-        model_name_query="consciousAI/question-answering-roberta-base-s-v2"
-        flag = False
-    elif genre == ':rainbow[deepset/roberta-base-squad2]':
-        st.write('You selected deepset/roberta-base-squad2')
-        model_name_query="deepset/roberta-base-squad2"
-        flag = False
-   
+    
 user_prompt = st.chat_input("Ask me anything about the content of the PDF:")
 if user_prompt:
     st.session_state.messages.append({'role': 'user', "content": user_prompt})