Update app.py

app.py

@@ -1,43 +1,25 @@
-from diffusers import StableDiffusionPipeline
 import torch
 import streamlit as st
+from diffusers import DiffusionPipeline
 
-# Set the page config as the very first command
-st.set_page_config(page_title="🔥 Memeify - Text-to-Meme Generator 🔥", page_icon="🔥", layout="wide")
-
-# Load the model
-model_id = "CompVis/stable-diffusion-v1-4"
-device = "cuda"
-
-pipe = StableDiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float16)
-pipe = pipe.to(device)
-
-# Custom styling for the page
-st.markdown("""
-    </style>
-""", unsafe_allow_html=True)
+# Load the pre-trained Stable Diffusion model
+pipe = DiffusionPipeline.from_pretrained("CompVis/stable-diffusion-v1-4")
 
 # Title and description
-st.markdown('<div class="title">🔥 MemeForge - Text-to-Meme Generator 🔥</div>', unsafe_allow_html=True)
+st.markdown('<div class="title">🔥 Memeify - Text-to-Meme Generator 🔥</div>', unsafe_allow_html=True)
 st.markdown('<div class="description">Enter a description, and the AI will generate a meme image for you!</div>', unsafe_allow_html=True)
 
 # User input for meme description
-input_text = st.text_input("Enter meme description:", key="input", placeholder="e.g., A funny dog looking confused.")
+input_text = st.text_input("Enter meme description:", key="input", placeholder="e.g., dog dancing")
 
 # Generate meme image when button is clicked
 if st.button("Generate Meme", key="generate", help="Click to generate a meme from your description"):
     if input_text:
         modified_prompt = "MEME of an Image of " + input_text
-
-        # Show a loading spinner while the image is being generated
-        with st.spinner('Generating your meme...'):
-            # Generate image from the prompt
-            image = pipe(modified_prompt, num_inference_steps=10).images[0]
-
-            # Display the generated image
-            st.image(image, caption="Your Generated Meme", use_column_width=True)
-
-            # Save the image
-            image.save("generated_meme.png")
+        # 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.")
+        st.warning("Please enter a description to generate a meme.")