Update app.py

app.py

@@ -4,14 +4,14 @@ from diffusers import DiffusionPipeline
 # Load the diffusion pipeline model
 @st.cache_resource
 def load_pipeline():
-    # Use the 'SaiRaj03/Text_To_Image' model for fast generation
-    pipe = DiffusionPipeline.from_pretrained("SaiRaj03/Text_To_Image")
+    # Using the 'black-forest-labs/FLUX.1-schnell' model for fast generation
+    pipe = DiffusionPipeline.from_pretrained("black-forest-labs/FLUX.1-schnell")
     return pipe
 
 pipe = load_pipeline()
 
 # Streamlit app
-st.title("Text-to-Image Generation App (Fast)")
+st.title("Text-to-Image Generation App (FLUX Model)")
 
 # User input for prompt
 user_prompt = st.text_input("Enter your image prompt", value="Astronaut in a jungle, cold color palette, muted colors, detailed, 8k")
@@ -20,10 +20,10 @@ user_prompt = st.text_input("Enter your image prompt", value="Astronaut in a jun
 if st.button("Generate Image"):
     if user_prompt:
         with st.spinner("Generating image..."):
-            # Generate the image using the new model
+            # Generate the image using the FLUX model
             image = pipe(user_prompt).images[0]
             
             # Display the generated image
-            st.image(image, caption="Generated Image", use_column_width=True)
+            st.image(image, caption="Generated Image (FLUX Model)", use_column_width=True)
     else:
         st.error("Please enter a valid prompt.")