Update app.py

app.py

@@ -31,8 +31,8 @@ clicked = st.button("Generate my movie")
 
 if clicked:
     generated_plot = model.generate_plot(prompt)[0]
-    chat_message = st.chat_message("assistant")
-    chat_message.markdown(generated_plot)
+    st.write("Assistant:")
+    st.markdown(generated_plot)
 
 with open('./style.css') as f:
     css = f.read()