Update app.py

app.py

@@ -157,9 +157,10 @@ def load_image(selected_file):
 def generate_input_image(image_path,input_choice):
     print(f"image_path: '{image_path}'")
     print(f"input_choice: '{input_choice}'")
-    if input_choice == 0 :
+    # ["Edge image","Segmented image","Depth Image"]
+    if input_choice == "Edge image" :
         conditioning_image = generate_edgeImage(image_path)
-    elif input_choice == 1 :
+    elif input_choice == "Segmented image" :
         conditioning_image = generate_segmt_img(image_path)
     else:
         conditioning_image = extract_depth_map(image_path)
@@ -167,9 +168,9 @@ def generate_input_image(image_path,input_choice):
     
 def generate_output_image(prompt,input_choice):
     #---------------------
-    if input_choice == 0 :
+    if input_choice == "Edge image":
         pipe = pipe_edge
-    elif input_choice == 1 :
+    elif input_choice == "Segmented image" :
         pipe = pipe_seg
     else:
         pipe = pipe_depth
@@ -255,7 +256,7 @@ with gr.Blocks() as demo:
     #)
     generate_inp_btn.click(
         fn=generate_input_image,
-        inputs=[file_input,image_dropdown],
+        inputs=[file_input[0],image_dropdown],
         outputs=[input_preview]
     )
     generate_btn.click(