Update app.py

app.py

@@ -43,7 +43,7 @@ with st.expander("Instructions"):
 
 option = st.sidebar.selectbox(
     "Which task do you want to do?",
-    ("Sentiment Analysis", "Medical Summarization", "Llama2 on YSA", "ChatGPT", "ChatGPT (with Google)"),
+    ("Sentiment Analysis", "Medical Summarization", "Llama2 on YSA", "Llama2 on BRK Letters", "ChatGPT", "ChatGPT (with Google)"),
 )
 
 
@@ -98,6 +98,20 @@ if prompt := st.chat_input("What is up?"):
                     final_response = call_chatgpt(query=engineered_prompt)
                 except:
                     final_response = "Sorry, the inference endpoint is temporarily down. 😔"
+        elif option == "Llama2 on BRK Letters":
+            if prompt:
+                try:
+                    out = llama2_7b_brk_letters(prompt)
+                    engineered_prompt = f"""
+                        The user asked the question: {prompt}
+    
+                        We have found relevant content: {out}
+    
+                        Answer the user question based on the above content in paragraphs.
+                    """
+                    final_response = call_chatgpt(query=engineered_prompt)
+                except:
+                    final_response = "Sorry, the inference endpoint is temporarily down. 😔"
         elif option == "ChatGPT":
             if prompt:
                 out = call_chatgpt(query=prompt)