Update app.py

app.py

@@ -39,7 +39,7 @@ class ChatbotUI:
 
         # Generate response using the model
         inputs = self.trainer.tokenizer(prompt, return_tensors="pt").input_ids
-        output = self.trainer.model.generate(inputs, max_length=1000, do_sample=True, top_p=0.95, temperature=0.6, repetition_penalty=1.2,)
+        output = self.trainer.model.generate(**inputs, max_length=1000, do_sample=True, top_p=0.95, temperature=0.6, repetition_penalty=1.2,)
         response = self.trainer.tokenizer.decode(output[0], skip_special_tokens=True)
         logging.info("Response generated successfully.")