Update app.py

app.py

@@ -1,31 +1,28 @@
-# Import necessary libraries
 import streamlit as st
-import torch
-from diffusers import StableDiffusionXLPipeline
+from diffusers import DiffusionPipeline
+import time
 
-# Load the stable diffusion model
-pipe = StableDiffusionXLPipeline.from_pretrained(
-    "stabilityai/stable-diffusion-xl-base-1.0", torch_dtype=torch.float16
-)
-pipe = pipe.to("cpu")  # Move the model to CPU for processing
+# Load the DiffusionPipeline model
+pipeline = DiffusionPipeline.from_pretrained("stabilityai/sdxl-turbo")
 
-# Create a Streamlit app
-st.title("Stable Diffusion XL Image Generation")
+# Function to generate and display the image
+def generate_and_display_image(prompt):
+    # Generate an image based on the prompt
+    image = pipeline(prompt)
+    # Display the generated image
+    st.image(image, caption='Generated Image', use_column_width=True)
 
-# Add a text input for the prompt
-prompt = st.text_area("Enter your prompt here", "Type your prompt here...")
+# Main app code
+st.title('Real-time Image Generation App')
 
-# Add an info text to provide guidance
-st.markdown("This app uses Stable Diffusion XL to generate an image based on the given prompt.")
+# User input for the prompt with a placeholder
+prompt = st.text_input('Enter your prompt', value='', help='Enter your prompt here...')
 
-# Generate the image based on the prompt
-if st.button("Generate"):
-    with st.spinner("Generating..."):  # Display a spinner while generating the image
-        try:
-            # Use the model to generate the image
-            image = pipe(prompt).images[0]
+# Initialize the image display
+image_placeholder = st.empty()
 
-            # Display the generated image
-            st.image(image, caption="Generated Image", use_column_width=True)
-        except Exception as e:
-            st.error(f"An error occurred: {e}")
+# Continuously generate and display the image based on the prompt
+while True:
+    generate_and_display_image(prompt)
+    # Sleep for a short duration to avoid excessive CPU usage
+    time.sleep(1)