Update main.py

main.py

@@ -22,7 +22,7 @@ def format_prompt(message, history):
 @app.post("/api/v1/generate_text")
 def generate_text(request: Request, prompt: str = Body()):
     history = []  # You might need to handle this based on your actual usage
-    # print(f"prompt + {prompt}")
+    print(f"request + {request}")
     temperature = request.headers.get("temperature", 0.5)
     # print(f"temperature + {temperature}")
     top_p = request.headers.get("top_p", 0.95)
@@ -50,6 +50,6 @@ def generate_text(request: Request, prompt: str = Body()):
         output += response.token.text
         yield output
         
-    return output
+    return output[-2]
 
     # return stream