Update app.py

app.py

@@ -18,6 +18,8 @@ import math
 from huggingface_hub import hf_hub_download
 from safetensors.torch import load_file
 from typing import Union, Tuple
+
+
 from basicsr.archs.rrdbnet_arch import RRDBNet
 from basicsr.utils.download_util import load_file_from_url
 from realesrgan import RealESRGANer
@@ -296,7 +298,7 @@ def infer(
     image_url=None,
     return_upscaled=False, 
     no_background=False, 
-    num_images_per_prompt=1,
+    num_images_per_prompt=4,
     progress=gr.Progress(track_tqdm=True),
 ):
     """
@@ -308,7 +310,8 @@ def infer(
     rewrite_prompt=False
     if randomize_seed:
         seed = random.randint(0, MAX_SEED)
-
+    if return_upscaled or no_background:
+        num_images_per_prompt = 1
     # Set up the generator for reproducibility
     generator = torch.Generator(device=device).manual_seed(seed)
     expected_key = os.environ.get("deepseek_key")
@@ -405,14 +408,14 @@ with gr.Blocks(css=css) as demo:
                                           show_label=False, 
                                           type="pil", 
                                           interactive=True)
-                image_url = gr.Textbox(label="", placeholder="")
+                image_url = gr.Textbox(label="option", placeholder="")
                 prompt = gr.Text(
                     label="Prompt",
                     show_label=True,
                     placeholder="",
                 )
-                return_upscaled = gr.Checkbox(label="", value=False)
-                remove_background = gr.Checkbox(label="", value=False)
+                return_upscaled = gr.Checkbox(label="upscale", value=False)
+                remove_background = gr.Checkbox(label="background remove", value=False)
                 run_button = gr.Button("Edit!", variant="primary")
                 
                 with gr.Accordion("Advanced Settings", open=False):
@@ -469,7 +472,7 @@ with gr.Blocks(css=css) as demo:
 
             with gr.Column():
                 result = gr.Gallery(label="", show_label=False, type="pil")
-                upscaled = gr.Image(label="")
+                upscaled = gr.Image(label="upscaled")
 
 
     gr.on(