Update app.py

app.py

@@ -503,15 +503,16 @@ model_path = snapshot_download(
     token=huggingface_token, # type a new token-id.
 )
 
-
 # Load pipeline
 controlnet = FluxControlNetModel.from_pretrained(
     "jasperai/Flux.1-dev-Controlnet-Upscaler", torch_dtype=torch.bfloat16
 ).to(device)
-pipe = FluxControlNetPipeline.from_pretrained(
+pipe_controlnet = FluxControlNetPipeline.from_pretrained(
     model_path, controlnet=controlnet, torch_dtype=torch.bfloat16
 )
-pipe.to(device)
+pipe_controlnet.to(device)
+
+
 
 MAX_SEED = 1000000
 
@@ -548,7 +549,8 @@ def process_input(input_image, upscale_factor):
     return input_image.resize((w, h)), w_original, h_original, was_resized
 
 MAX_PIXEL_BUDGET = 1024 * 1024
-    
+
+
 @spaces.GPU
 def upscale(input_image, progress=gr.Progress(track_tqdm=True)):
     if input_image is None:
@@ -563,7 +565,7 @@ def upscale(input_image, progress=gr.Progress(track_tqdm=True)):
     generator = torch.Generator(device=device).manual_seed(random.randint(0, MAX_SEED))
 
     gr.Info("Upscaling image to 4096x4096...")
-    image = pipe_controlnet(
+    upscaled_image = pipe_controlnet(
         prompt="",
         image=control_image,
         controlnet_conditioning_scale=0.6,
@@ -574,7 +576,8 @@ def upscale(input_image, progress=gr.Progress(track_tqdm=True)):
         generator=generator,
     ).images[0]
 
-    return image
+    return [input_image, upscaled_image]
+
     
 
 with gr.Blocks(theme="Nymbo/Nymbo_Theme", css=css, delete_cache=(60, 3600)) as app:
@@ -625,11 +628,18 @@ with gr.Blocks(theme="Nymbo/Nymbo_Theme", css=css, delete_cache=(60, 3600)) as a
                 columns=4,
                 elem_id="gallery"
             )
-        with gr.Column():
-            progress_bar = gr.Markdown(elem_id="progress", visible=False)
-            result = gr.Image(label="Generated Image", interactive=False)
-            with gr.Accordion("History", open=False):
-                history_gallery = gr.Gallery(label="History", columns=6, object_fit="contain", interactive=False)
+    with gr.Column():
+        progress_bar = gr.Markdown(elem_id="progress", visible=False)
+        result = ImageSlider(
+            label="Generated Image",
+            minimum=0,
+            maximum=100,
+            step=1,
+            value=50,
+            elem_id="result_slider"
+        )
+        with gr.Accordion("History", open=False):
+            history_gallery = gr.Gallery(label="History", columns=6, object_fit="contain", interactive=False)
 
     with gr.Row():
         with gr.Accordion("Advanced Settings", open=False):
@@ -689,6 +699,7 @@ with gr.Blocks(theme="Nymbo/Nymbo_Theme", css=css, delete_cache=(60, 3600)) as a
         outputs=history_gallery,
     )
     
+
     upscale_button.click(
         upscale,
         inputs=[result],