Updates

app.py

@@ -18,43 +18,68 @@ def initialize_model():
 def generate_image(
     prompt,
     guidance_scale=3.5,
-    num_steps=25,  # Changed default to 25
     width=1024,
     height=1024
 ):
-    # Initialize model within the GPU context
-    pipe = initialize_model()
-    
-    # Generate random seed
-    seed = random.randint(1, 1000000)
-    
-    with torch.inference_mode():
-        image = pipe(
-            prompt=prompt,
-            generator=torch.Generator(device="cuda").manual_seed(seed),
-            num_inference_steps=num_steps,
-            guidance_scale=guidance_scale,
-            height=height,
-            width=width,
-        ).images[0]
-    
-    return image
+    try:
+        # Initialize model within the GPU context
+        pipe = initialize_model()
+        
+        # Generate random seed
+        seed = random.randint(1, 1000000)
+        
+        with torch.inference_mode():
+            image = pipe(
+                prompt=prompt,
+                generator=torch.Generator(device="cuda").manual_seed(seed),
+                num_inference_steps=25,  # Fixed steps
+                guidance_scale=guidance_scale,
+                height=height,
+                width=width,
+            ).images[0]
+        
+        return image
+    except Exception as e:
+        print(f"Error during image generation: {str(e)}")
+        raise e
 
 # Create the Gradio interface
 demo = gr.Interface(
     fn=generate_image,
     inputs=[
-        gr.Textbox(label="Prompt", placeholder="Enter your image description here..."),
-        gr.Slider(minimum=1, maximum=20, value=3.5, label="Guidance Scale", step=0.5),
-        gr.Slider(minimum=128, maximum=1024, value=1024, label="Width", step=64),
-        gr.Slider(minimum=128, maximum=1024, value=1024, label="Height", step=64)
+        gr.Textbox(
+            label="Prompt",
+            placeholder="Enter your image description here...",
+            value="A serene landscape with mountains at sunset"
+        ),
+        gr.Slider(
+            minimum=1,
+            maximum=20,
+            value=3.5,
+            label="Guidance Scale",
+            step=0.5
+        ),
+        gr.Slider(
+            minimum=128,
+            maximum=1024,
+            value=1024,
+            label="Width",
+            step=64
+        ),
+        gr.Slider(
+            minimum=128,
+            maximum=1024,
+            value=1024,
+            label="Height",
+            step=64
+        )
     ],
     outputs=gr.Image(type="pil", label="Generated Image"),
     title="Flux Image Generator (Zero-GPU)",
-    description="Generate images using Freepik's Flux model with Zero-GPU allocation. Using 25 steps and random seed for each generation.",
+    description="Generate images using Freepik's Flux model with Zero-GPU allocation. Using 25 fixed steps and random seed for each generation.",
     examples=[
         ["A close-up image of a green alien with fluorescent skin in the middle of a dark purple forest", 3.5, 1024, 1024],
-        ["A serene landscape with mountains at sunset", 3.5, 1024, 1024],
+        ["A serene landscape with mountains at sunset", 3.5, 1024, 1024]
     ]
 )