Update app/main.py

app/main.py

@@ -41,12 +41,16 @@ async def chat(message: ChatMessage):
     
     # Generate response using pipeline
     response = pipe(messages)
+
+    print(response)
     
     # Extract the response text
     if isinstance(response, list):
         response_text = response[0].get('generated_text', '')
+        print(response_text)
     else:
         response_text = response.get('generated_text', '')
+        print(response_text)
     
     return {"response": response_text}