Update app.py

app.py

@@ -21,4 +21,9 @@ if st.button("Generate Meme", key="generate", help="Click to generate a meme fro
     if input_text:
         modified_prompt = "MEME of an Image of " + input_text
         # Generate the image from the modified prompt
-
+        image = pipe(modified_prompt).images[0]
+        
+        # Display the generated image in the Streamlit app
+        st.image(image, caption=f"Meme: {input_text}", use_column_width=True)
+    else:
+        st.warning("Please enter a description to generate a meme.")