Update app.py

app.py

@@ -116,7 +116,7 @@ async def generate(gen:GenModel):
         """
         et = time()
         #output["time"] = et - st
-        return output['choices'][0]
+        return output
     except Exception as e:
         logger.error(f"Error in /generate endpoint: {e}")
         return JSONResponse(