Update app.py

app.py

@@ -103,6 +103,8 @@ if 1:
                 out_gens = output_text
                 prompt = ""
             st.balloons()
+
+            out_gens = out_gens.split('\n')[0]
             
             def multiline(string):
                 lines = string.split('\n')