Update app.py

app.py

@@ -38,10 +38,10 @@ def respond(
 
     response = ""
 
-    for message in client(
+    
+    token = client(
         messages, max_new_tokens= max_tokens
-    ):
-        token = message
+    )
         print(token)
         response += token
         yield response