Update app.py

app.py

@@ -152,8 +152,13 @@ def generate(
     )
 
     formatted_prompt = format_prompt(f"{system_prompt}\n\n{prompt}", history)
-    output = client.text_generation(formatted_prompt, **generate_kwargs, stream=False, return_full_text=False)
-    
+
+    try:
+        output = client.text_generation(formatted_prompt, **generate_kwargs, stream=False, return_full_text=False)
+    except Exception as e:
+        logger.error(f"Error during model inference: {e}")
+        return "Sorry, there was an error processing your request. Please try again later."
+
     return output
 
 def update_sys_prompt(agent):