Update app.py

app.py

@@ -22,7 +22,7 @@ print("Start the model init process")
 model = GPT4All(model_name=model_name,
                 model_path=model_path,
                 allow_download = False, 
-                evice="cpu")
+                device="cpu")
 print("Finish the model init process")
 
 def generater(message, history):
@@ -37,12 +37,11 @@ def generater(message, history):
                                 streaming=True):
         outputs.append(token)
         completion = "".join(outputs)
+        yield completion
         if completion.endswith(" [eod]"):
-            transferred_completion = utils.transfer_llm_completion(completion, message)
-            yield transferred_completion
-            return
-        else:
-            yield completion
+            break
+    transferred_completion = utils.transfer_llm_completion(completion, message)
+    yield transferred_completion
         
 def vote(data: gr.LikeData):
     if data.liked: