Update app.py

app.py

@@ -70,7 +70,7 @@ with st.sidebar:
 
     # Select FM
     option = st.selectbox(
-        "Which foundational model would you like?", ("GPT4", "LLAMA3")
+        "Which foundational model would you like?", ("GPT4", "LLAMA3", "LLAMA2")
     )
 
     # Clear button
@@ -139,16 +139,17 @@ elif uploaded_files:
 
             # Call FM
             content = " ".join(list(result.sentences))
+            custom_prompt = f"""
+                Answer the question: {prompt} 
+                
+                Use the following information: {content}
+                """
             if option == "GPT4":
                 response = call_gpt(prompt, content)
+            elif option =="LLAMA2":
+                response = call_llama2(custom_prompt)
             else:
-                response = call_llama(
-                    f"""
-                    Answer the question: {prompt} 
-                    
-                    Use the following information: {content}
-                    """
-                )
+                response = call_llama(custom_prompt)
 
             # Display assistant response in chat message container
             with st.chat_message("assistant"):