Update app.py

app.py

@@ -1,38 +1,32 @@
 import torch
-import gradio as gr
 from diffusers import StableDiffusion3Pipeline
+import gradio as gr
 
-def image_generator(prompt):
-    device = "cuda" if torch.cuda.is_available() else "cpu"
-    pipeline = StableDiffusion3Pipeline.from_pretrained("stabilityai/stable-diffusion-3-medium-diffusers", 
-                                                         torch_dtype=torch.float16 if device == "cuda" else torch.float32,
-                                                         text_encoder_3=None,
-                                                         tokenizer_3 = None)
-    #pipeline.enable_model_cpu_offload()
-    pipeline.to(device)
-
-    image = pipeline(
-        prompt=prompt,
-        negative_prompt="blurred, ugly, watermark, low, resolution, blurry",
-        num_inference_steps=40,
-        height=1024,
-        width=1024,
-        guidance_scale=9.0
+# Load the Stable Diffusion pipeline
+pipe = StableDiffusion3Pipeline.from_pretrained(
+    "stabilityai/stable-diffusion-3-medium-diffusers",
+    torch_dtype=torch.float16
+)
+pipe = pipe.to("cuda")
+
+# Function to generate images
+def generate_image(prompt):
+    image = pipe(
+        prompt,
+        negative_prompt="",
+        num_inference_steps=28,
+        guidance_scale=7.0,
     ).images[0]
-
-    #image.show()
     return image
 
-#image_generator("A magical cat doing spell")
-
-interface = gr.Interface(
-    fn=image_generator, 
-    inputs=gr.Textbox(lines=2, placeholder = "Enter your prompt..."), 
-    outputs=gr.Image(type = "pil"),
-    title = "Image Generator App",
-    description = "This is a simple image generator app using HuggingFace's Stable Diffusion 3 model.")
-
-interface.launch()
-
-
-print(interface)
+# Create a Gradio interface
+iface = gr.Interface(
+    fn=generate_image,                       # Function to call
+    inputs=gr.Textbox(label="Enter a prompt"),  # Input component
+    outputs=gr.Image(label="Generated Image"),   # Output component
+    title="Stable Diffusion 3 Image Generator",
+    description="Enter a prompt to generate an image using Stable Diffusion 3."
+)
+
+# Launch the Gradio app
+iface.launch()