Update app.py

app.py

@@ -148,26 +148,25 @@ def resize_img(control_image):
     return resized_image
 
 @spaces.GPU(duration=180)
-def infer(cond_in, image_in, prompt, inference_steps, guidance_scale, control_mode, control_strength, seed, progress=gr.Progress(track_tqdm=True)):
+def infer(image_in, prompt, inference_steps, guidance_scale, control_mode, control_strength, seed, progress=gr.Progress(track_tqdm=True)):
     control_mode_num = mode_mapping[control_mode]
     
-    if cond_in is None:
-        if image_in is not None:
-            image_in = resize_img(load_image(image_in))
-            if control_mode == "canny":
-                control_image = extract_canny(image_in)
-            elif control_mode == "depth":
-                control_image = extract_depth(image_in)
-            elif control_mode == "pose":
-                control_image = extract_openpose(image_in)
-            elif control_mode == "colorgrid":
-                control_image = tile(64, image_in)
-            elif control_mode == "recolor":
-                control_image = convert_to_grayscale(image_in)
-            elif control_mode == "tile":
-                control_image = tile(16, image_in)
-    else:
-        control_image = resize_img(load_image(cond_in))
+    if image_in is not None:
+        image_in = resize_img(load_image(image_in))
+        if control_mode == "canny":
+            control_image = extract_canny(image_in)
+        elif control_mode == "depth":
+            control_image = extract_depth(image_in)
+        elif control_mode == "pose":
+            control_image = extract_openpose(image_in)
+        elif control_mode == "colorgrid":
+            control_image = tile(64, image_in)
+        elif control_mode == "recolor":
+            control_image = convert_to_grayscale(image_in)
+        elif control_mode == "tile":
+            control_image = tile(16, image_in)
+
+    control_image = resize_img(control_image)
 
     width, height = control_image.size
     
@@ -257,7 +256,7 @@ with gr.Blocks(css=css) as demo:
         api_name=False
     ).then(
         fn = infer,
-        inputs = [None, image_in, prompt, inference_steps, guidance_scale, control_mode, control_strength, seed],
+        inputs = [image_in, prompt, inference_steps, guidance_scale, control_mode, control_strength, seed],
         outputs = [result, processed_cond],
         show_api=False
     )