Update app.py

app.py

@@ -38,17 +38,16 @@ def complete(prompt, stop=["User", "Assistant"]):
         output += result
         for word in stop:
             if word in output:
-                print('\n')
-                return [output, token_count]
-        print(result, end='', flush=True)
+                return output, token_count
 
-    print('\n')
-    return [output, token_count]
+    return output, token_count
 
 def greet(question):
     print(question)
     output, token_count = complete(f'User: {question}. Can you please answer this as informatively but concisely as possible.\nAssistant: ')
-    return f"Response: {output} | Tokens: {token_count}"
+    response = f"Response: {output} | Tokens: {token_count}"
+    print(response)
+    return response
 
 iface = gr.Interface(fn=greet, inputs="text", outputs="text")
 iface.launch(share=True)