Update main.py

main.py

@@ -51,5 +51,5 @@ def generate(item: Item):
 @app.post("/generate/")
 async def generate_text(item: Item):
     ans = generate(item)
-    print(ans)
+    print(item.prompt)
     return {"response": ans}