Update app.py

app.py

@@ -35,7 +35,7 @@ MAX_SEED = np.iinfo(np.int32).max
 MAX_IMAGE_SIZE = 2048
 
 @spaces.GPU()
-def infer(prompt, seed=42, randomize_seed=False, width=1024, height=1024, num_inference_steps=4, num_feifei=0.35, progress=gr.Progress(track_tqdm=True)):
+def infer(prompt, seed=42, randomize_seed=False, width=1024, height=1024, num_inference_steps=4, guidancescale=3.5, num_feifei=0.35, progress=gr.Progress(track_tqdm=True)):
     if randomize_seed:
         seed = random.randint(0, MAX_SEED)
     generator = torch.Generator().manual_seed(seed)
@@ -55,7 +55,7 @@ def infer(prompt, seed=42, randomize_seed=False, width=1024, height=1024, num_in
             height = height,
             num_inference_steps = num_inference_steps, 
             generator = generator,
-            guidance_scale=3.5
+            guidance_scale=guidancescale
     ).images[0] 
     return image, seed
  
@@ -133,7 +133,6 @@ with gr.Blocks(css=css) as demo:
                 )
             
             with gr.Row():
-                
                 num_inference_steps = gr.Slider(
                     label="Number of inference steps",
                     minimum=1,
@@ -141,6 +140,15 @@ with gr.Blocks(css=css) as demo:
                     step=1,
                     value=8,
                 )
+
+            with gr.Row():
+                guidancescale = gr.Slider(
+                    label="Guidance scale",
+                    minimum=0,
+                    maximum=10,
+                    step=0.1,
+                    value=3.5,
+                )
                 
             with gr.Row():
                 num_feifei = gr.Slider(
@@ -162,7 +170,7 @@ with gr.Blocks(css=css) as demo:
     gr.on(
         triggers=[run_button.click, prompt.submit],
         fn = infer,
-        inputs = [prompt, seed, randomize_seed, width, height, num_inference_steps, num_feifei],
+        inputs = [prompt, seed, randomize_seed, width, height, num_inference_steps, guidancescale, num_feifei],
         outputs = [result, seed]
     )