Update app.py

app.py

@@ -1,5 +1,3 @@
-
-
 #import gradio as gr
 #gr.load("models/mistralai/Mistral-7B-Instruct-v0.3").launch()
 
@@ -20,9 +18,12 @@ def query(payload):
 
 def chatbot_response(input_text):
     response = query({"inputs": input_text})
-    if 'error' in response:
-        return response['error']
-    return response.get('generated_text', 'No response generated.')
+    # Extract the generated text from the response
+    if isinstance(response, dict) and 'generated_text' in response:
+        return response['generated_text']
+    elif isinstance(response, list) and len(response) > 0 and 'generated_text' in response[0]:
+        return response[0]['generated_text']
+    return 'No response generated.'
 
 # Gradio interface
 def main():