updated main.py to give response in JSON format

main.py

@@ -27,7 +27,7 @@ def format_prompt(message, history):
     prompt += f"[INST] {message} [/INST]"
     return prompt
 
-@app.post("/api/v1/generate_text")
+@app.post("/api/v1/generate_text", response_model=dict)
 def generate_text(
     request: Request,
     body: dict = Body(...),
@@ -56,4 +56,4 @@ def generate_text(
         return_full_text=False
     )
 
-    return stream
+    return {"generated_text": stream}