Update app.py

app.py

@@ -2,26 +2,27 @@ import gradio as gr
 from diffusers import AutoPipelineForText2Image
 import torch
 
-# Load sd-turbo (fast and lightweight)
+# Load the sd-turbo model
 pipe = AutoPipelineForText2Image.from_pretrained(
     "stabilityai/sd-turbo",
-    torch_dtype=torch.float32  # Use float32 for CPU
+    torch_dtype=torch.float32  # Use float32 for CPU (or float16 for GPU)
 ).to("cuda" if torch.cuda.is_available() else "cpu")
 
-# Optional warm-up (to reduce first call delay)
-pipe("a cat wearing sunglasses")
+# Warm-up to reduce first-time delay (optional)
+pipe("a test prompt")
 
+# Gradio interface (simple)
 def generate_image(prompt):
     image = pipe(prompt).images[0]
     return image
 
-# Gradio UI and API endpoint
+# Use only launch() here (no enable_queue)
 iface = gr.Interface(
     fn=generate_image,
     inputs="text",
     outputs="image",
-    title="Text to Image with SD-Turbo",
-    description="Enter a text prompt and generate an image using stabilityai/sd-turbo"
+    title="Text-to-Image with SD-Turbo",
+    description="Fast free text-to-image generation using stabilityai/sd-turbo"
 )
 
-iface.launch(share=True, enable_queue=True)
+iface.launch()