Update app.py

app.py

@@ -105,15 +105,18 @@ if prompt := st.chat_input("What is up?"):
                 final_response = out[0]["summary_text"]
         elif option == "Llama2 on YSA":
             if prompt:
-                out = llama2_7b_ysa(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)
+                try:
+                    out = llama2_7b_ysa(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 done."
         elif option == "ChatGPT":
             if prompt:
                 out = call_chatgpt(query=prompt)