Update app.py

app.py

@@ -35,10 +35,24 @@ except Exception as e:
 # Function to generate a response from the model
 def generate_response(user_input):
     try:
+        print(f"User input: {user_input}")
         messages = [{"role": "user", "content": user_input}]
-        input_ids = tokenizer.apply_chat_template(conversation=messages, tokenize=True, add_generation_prompt=True, return_tensors='pt')
+        print(f"Messages: {messages}")
+
+        input_ids = tokenizer.apply_chat_template(
+            conversation=messages,
+            tokenize=True,
+            add_generation_prompt=True,
+            return_tensors='pt'
+        )
+        print(f"Input IDs: {input_ids}")
+
         output_ids = model.generate(input_ids.to('cuda'))
+        print(f"Output IDs: {output_ids}")
+
         response = tokenizer.decode(output_ids[0][input_ids.shape[1]:], skip_special_tokens=True)
+        print(f"Response: {response}")
+
         return response
     except Exception as e:
         print(f"Error generating response: {e}")