Update app.py

app.py

@@ -86,9 +86,10 @@ def chat(message,
                 json_obj = json.loads(json_line)
                 if 'token' in json_obj:
                     partial_message = partial_message + json_obj['token']['text'] 
-                    yield partial_message
+                    return partial_message #yield
                 elif 'error' in json_obj:
-                    yield json_obj['error'] + '. Please refresh and try again with an appropriate smaller input prompt.'
+                    return json_obj['error'] + '. Please refresh and try again with an appropriate smaller input prompt.'
+                    # yield
                 else:
                     gr.Warning(f"The key 'token' does not exist in this JSON object: {json_obj}")