Update app.py

app.py

@@ -27,7 +27,7 @@ def extract_resolution(resolution_str):
 
 # Define the image generation function with adjustable parameters and a progress bar
 @spaces.GPU
-def generate(prompt, guidance_scale, num_inference_steps, resolution, negative_prompt):
+def generate(prompt, stage1_guidance_scale, stage2_guidance_scale, num_inference_steps, resolution, negative_prompt):
     width, height = extract_resolution(resolution) or (1024, 1024)
     mixture_generator = torch.Generator().manual_seed(444)
     stage1_strength = 0.6
@@ -37,7 +37,7 @@ def generate(prompt, guidance_scale, num_inference_steps, resolution, negative_p
         num_inference_steps=num_inference_steps,
         num_images_per_prompt=1,
         generator=mixture_generator,
-        guidance_scale=guidance_scale,
+        guidance_scale=stage1_guidance_scale,
         output_type="latent",
         denoising_end=stage1_strength,
         width=width,
@@ -50,15 +50,16 @@ def generate(prompt, guidance_scale, num_inference_steps, resolution, negative_p
         num_inference_steps=num_inference_steps,
         num_images_per_prompt=1,
         generator=mixture_generator,
-        guidance_scale=guidance_scale,
+        guidance_scale=stage2_guidance_scale,
         denoising_start=stage1_strength
     ).images
 
 # Example prompts to demonstrate the model's capabilities
 example_prompts = [
-    ["A futuristic cityscape at night under a starry sky", 7.5, 25, "blurry, overexposed"],
-    ["A serene landscape with a flowing river and autumn trees", 8.0, 20, "crowded, noisy"],
-    ["An abstract painting of joy and energy in bright colors", 9.0, 30, "dark, dull"]
+    ["A futuristic cityscape at night under a starry sky", 3.5, 4.5, 25, "blurry, overexposed"],
+    ["A serene landscape with a flowing river and autumn trees", 3.0, 4.0, 20, "crowded, noisy"],
+    ["An abstract painting of joy and energy in bright colors", 3.0, 4.5, 30, "dark, dull"],
+    ["a stunning portrait of a hamster with an eye patch, piloting a miniature cessna on a wooden desk in an office, depth of field, bokeh, sharp, f1.4", 3.2, 4.6, 40, "this is an ugly photograph that no one liked"]
 ]
 
 # Create a Gradio interface, 1024x1024,1152x960,896x1152
@@ -66,8 +67,9 @@ iface = gr.Interface(
     fn=generate,
     inputs=[
         gr.Text(label="Enter your prompt"),
-        gr.Slider(1, 20, step=0.1, label="Guidance Scale", value=3.4),
-        gr.Slider(1, 50, step=1, label="Number of Inference Steps", value=28),
+        gr.Slider(1, 20, step=0.1, label="Guidance Scale (Stage I)", value=3.4),
+        gr.Slider(1, 20, step=0.1, label="Guidance Scale (Stage II)", value=4.2),
+        gr.Slider(1, 50, step=1, label="Number of Inference Steps", value=35),
         gr.Radio(["1024x1024", "1152x960", "896x1152"], label="Resolution", value="1152x960"),
         gr.Text(value="underexposed, blurry, ugly, washed-out", label="Negative Prompt")
     ],