Update app.py

app.py

@@ -84,10 +84,11 @@ if st.button('Generate'):
     st.write("Generating....")
     text = ""
     i = 0
-    st.write(text)
-    for new_text in streamer:
-        i += 1
-        if (i != 1):
-            new_text = new_text.split("<|endoftext|>", 1)[0]
-            text += new_text
+    st.write_stream(streamer)
+    
+    #for new_text in streamer:
+        #i += 1
+        #if (i != 1):
+            #new_text = new_text.split("<|endoftext|>", 1)[0]
+            #text += new_text