Update app.py

app.py

@@ -24,7 +24,7 @@ if prompt:
         	"inputs": prompt,
         })
         if output != None:
-            user_msg = st.chat_message("User")
+            user_msg = cont.chat_message("User")
             user_msg.write(prompt)
-            as_msg = st.chat_message("Assistant")
-            as_msg.write(output)
+            as_msg = cont.chat_message("Assistant")
+            as_msg.write(output[0]['generated_text'])