Update app.py

app.py

@@ -10,7 +10,7 @@ STYLE_PRESETS = [None, "3d-model", "analog-film", "anime", "cinematic", "comic-b
 MAX_SEED = np.iinfo(np.int32).max
 
 
-def generate_image(upload_images, prompt, negative_prompt, style_preset, steps, strength, seed, progress=gr.Progress(track_tqdm=True)):
+def generate_image(upload_images, prompt, negative_prompt, style_preset, steps, cfg_scale, strength, seed, progress=gr.Progress(track_tqdm=True)):
     error_if_no_img(prompt)
 
     job = prodia.create("/photomaker",
@@ -19,6 +19,7 @@ def generate_image(upload_images, prompt, negative_prompt, style_preset, steps,
                         negative_prompt=negative_prompt,
                         style_preset=style_preset,
                         steps=steps,
+                        cfg_scale,
                         strength=strength,
                         seed=seed if seed != 0 else None
                         )
@@ -107,27 +108,35 @@ with gr.Blocks(css=css) as demo:
                     placeholder="low quality",
                     value="nsfw, lowres, bad anatomy, bad hands, text, error, missing fingers, extra digit, fewer digits, cropped, worst quality, low quality, normal quality, jpeg artifacts, signature, watermark, username, blurry",
                 )
-                steps = gr.Slider(
-                    label="Number of sample steps",
-                    minimum=20,
-                    maximum=50,
-                    step=1,
-                    value=40,
-                )
-                strength_ratio = gr.Slider(
-                    label="Strength (%)",
-                    minimum=15,
-                    maximum=50,
-                    step=1,
-                    value=20,
-                )
-                seed = gr.Slider(
-                    label="Seed",
-                    minimum=0,
-                    maximum=MAX_SEED,
-                    step=1,
-                    value=0,
-                )
+                with gr.Row():
+                    steps = gr.Slider(
+                        label="Number of sample steps",
+                        minimum=20,
+                        maximum=50,
+                        step=1,
+                        value=40,
+                    )
+                    cfg_scale = gr.Slider(
+                        label="CFG Scale",
+                        minimum=5,
+                        maximum=20,
+                        value=7,
+                    )
+                with gr.Row():
+                    strength_ratio = gr.Slider(
+                        label="Strength (%)",
+                        minimum=15,
+                        maximum=50,
+                        step=1,
+                        value=20,
+                    )
+                    seed = gr.Slider(
+                        label="Seed",
+                        minimum=0,
+                        maximum=MAX_SEED,
+                        step=1,
+                        value=0,
+                    )
         with gr.Column():
             result_image = gr.Image(label="Generated Image")
 
@@ -136,7 +145,7 @@ with gr.Blocks(css=css) as demo:
 
         submit.click(
             fn=generate_image,
-            inputs=[files, prompt, negative_prompt, style, steps, strength_ratio, seed],
+            inputs=[files, prompt, negative_prompt, style, steps, cfg_scale, strength_ratio, seed],
             outputs=[result_image]
         )