Update app.py

app.py

@@ -18,7 +18,7 @@ def generate_response(system_prompt, user_prompt):
         {"role": "user", "content": user_prompt},
     ]
     
-    output = pipe(text=messages, max_new_tokens=2048)
+    output = pipe(messages, max_new_tokens=2048)
     return output[0]["generated_text"][-1]["content"]
 
 # Create Gradio UI