Update app.py

app.py

@@ -93,6 +93,12 @@ def generate(
     
     return output
 
+def generate_norm(*args) -> str:
+    t = ""
+    for chunk in generate(*args):
+        t += chunk
+    return t
+
 @app.get('/')
 async def index():
     return JSONResponse({ "message": "hello", "url": "https://aweirddev-mistral-7b-instruct-v0-2-leicht.hf.space" })
@@ -112,7 +118,7 @@ async def c_cmp(payload: Payload):
                         "index": 0,
                         "message": {
                             "role": "assistant",
-                            "content": generate(
+                            "content": generate_norm(
                                 payload.model_dump()['messages'],
                                 payload.temperature,
                                 4096,