Update app.py

app.py

@@ -606,7 +606,6 @@ def infer_upscale(
 
         # resize to target desired size
         image = image.resize((w_original * upscale_factor, h_original * upscale_factor))
-        image.save("output.jpg")
         
         return image, seed
     except Exception as e:
@@ -769,20 +768,45 @@ with gr.Blocks(theme="Nymbo/Nymbo_Theme", css=css, delete_cache=(60, 3600)) as a
         inputs=[result, history_gallery],
         outputs=history_gallery,
     )
+
+
+def check_upscale_input(input_image, *args):
+    if input_image is None:
+        raise gr.Error("Please provide an input image for upscaling.")
+    return input_image, *args
+
+upscale_button.click(
+    check_upscale_input,
+    inputs=[
+        upscale_input,
+        upscale_seed,
+        upscale_randomize_seed,
+        upscale_steps,
+        upscale_factor,
+        controlnet_conditioning_scale,
+    ],
+    outputs=[
+        upscale_input,
+        upscale_seed,
+        upscale_randomize_seed,
+        upscale_steps,
+        upscale_factor,
+        controlnet_conditioning_scale,
+    ]
+).then(
+    infer_upscale,
+    inputs=[
+        upscale_seed,
+        upscale_randomize_seed,
+        upscale_input,
+        upscale_steps,
+        upscale_factor,
+        controlnet_conditioning_scale,
+    ],
+    outputs=[upscale_result, upscale_seed_output],
+)
+
     
-    # 업스케일 버튼 이벤트 추가
-    upscale_button.click(
-        infer_upscale,
-        inputs=[
-            upscale_seed,
-            upscale_randomize_seed,
-            upscale_input,
-            upscale_steps,
-            upscale_factor,
-            controlnet_conditioning_scale,
-        ],
-        outputs=[upscale_result, upscale_seed_output],
-    )
 
 if __name__ == "__main__":
     app.queue(max_size=20)