Update app.py

app.py

@@ -13,7 +13,10 @@ def chat_with_model(history):
     prompt = history[-1]["content"]
     try:
         response = generator(prompt)
-        result = response["generated_text"] if "generated_text" in response else response
+        if isinstance(response, dict) and "generated_text" in response:
+            result = response["generated_text"]
+        else:
+            result = response
     except Exception as e:
         result = f"Error: {str(e)}"
     return history + [{"role": "bot", "content": result}]