Update app.py

app.py

@@ -17,15 +17,15 @@ def get_flux_image(prompt):
     print(result)
     return result[0]
 
-def get_upscale(prompt, img_path):
+def get_upscale(prompt, img_path, upscale_factor):
     client = Client("finegrain/finegrain-image-enhancer")
     result = client.predict(
 		input_image=handle_file(img_path),
 		prompt=prompt,
 		negative_prompt="",
 		seed=42,
-		upscale_factor=2,
-		controlnet_scale=0.6,
+		upscale_factor=upscale_factor,
+	    controlnet_scale=0.6,
 		controlnet_decay=1,
 		condition_scale=6,
 		tile_width=112,
@@ -38,9 +38,9 @@ def get_upscale(prompt, img_path):
     print(result)
     return result[1]
 
-def main(prompt):
+def main(prompt, upscale_factor):
     step_one_flux = get_flux_image(prompt)
-    step_two_upscale = get_upscale(prompt, step_one_flux)
+    step_two_upscale = get_upscale(prompt, step_one_flux, upscale_factor)
     return (step_one_flux, step_two_upscale)
 
 css = """
@@ -51,14 +51,22 @@ css = """
 """
 with gr.Blocks(css=css) as demo:
     with gr.Column(elem_id="col-container"):
-        gr.Markdown("Flux Upscaled")
+        gr.Markdown("# Flux Upscaled")
         prompt_in = gr.Textbox(label="Prompt")
-        submit_btn = gr.Button("Submit")
+        with gr.Row():
+            upscale_factor = gr.Radio(
+                label = "UpScale Factor",
+                choices = [
+                    2, 3, 4
+                ],
+                value = 2
+            )
+            submit_btn = gr.Button("Submit")
         output_res = ImageSlider(label="Flux / Upscaled")
 
     submit_btn.click(
         fn=main,
-        inputs=[prompt_in],
+        inputs=[prompt_in, upscale_factor],
         outputs=[output_res],
         
     )