return type

app.py

@@ -41,7 +41,7 @@ def generate(text: str):
     # print(output)
     
     print(output)
-    return {"output": output[0]["generated_text"][-1]['content']}
+    
 
 @app.post("/chat")
 def chat(request: ChatRequest):
@@ -50,4 +50,4 @@ def chat(request: ChatRequest):
     """
     
     output = pipe(request.conversationHistory, num_return_sequences=1, max_new_tokens=250)
-    return output
+    return {"output": output[0]["generated_text"][-1]['content']}