Update app.py

app.py

@@ -60,7 +60,7 @@ with st.form(key='image_form'):
 if generate_image_button and st.session_state['dynamic_prompt']:
     # Generate the image using your custom model
     # Assuming 'my_custom_model_generate_image' is a function that returns the image data
-    st.session_state['image_data'] = my_custom_model_generate_image(st.session_state['dynamic_prompt'])
+    st.session_state['image_data'] = (st.session_state['dynamic_prompt'])
 
     # Display the generated image if available
     if st.session_state['image_data'] is not None: