Update app.py

app.py

@@ -71,6 +71,8 @@ def get_tryon_result(model_name, garment1, garment2, seed=1234):
     # model_name = "AI Model " + model_name.split("\\")[-1].split(".")[0] # windows
     # model_name = "AI Model " + model_name.split("/")[-1].split(".")[0] # linux
     # print(model_name)
+    garment1 = cv2.imread(garment1)  # read path into numpy
+    garment1 = cv2.cvtColor(garment1, cv2.COLOR_BGR2RGB)
 
     if not model_name.startswith("AI Model "):
         model_name = "AI Model " + model_name.split("/")[-1].split(".")[0]
@@ -201,8 +203,8 @@ with gr.Blocks(css = ".output-image, .input-image, .image-preview {height: 400px
                 bottom_btn4 = gr.Button("Shorts", variant="secondary")
 
             with gr.Row():
-                garment_top = gr.Image(sources='upload', type="numpy", label="top garment")
-                garment_down = gr.Image(sources='upload', type="numpy", label="lower garment")
+                garment_top = gr.Image(sources='upload', type="filepath", label="top garment")
+                garment_down = gr.Image(sources='upload', type="filepath", label="lower garment")
 
             run_button = gr.Button(value="Run")
             
@@ -230,36 +232,36 @@ with gr.Blocks(css = ".output-image, .input-image, .image-preview {height: 400px
 
     # Connect buttons to load sample images
     top_btn1.click(
-        lambda: load_sample_image(SAMPLE_CLOTHES["top"][0]),
+        lambda: load_sample_image(SAMPLE_CLOTHES["top"][0], outputs=[garment_top]),
         outputs=[garment_top]
     )
     top_btn2.click(
-        lambda: load_sample_image(SAMPLE_CLOTHES["top"][1]), 
+        lambda: load_sample_image(SAMPLE_CLOTHES["top"][1], outputs=[garment_top]), 
         outputs=[garment_top]
     )
     top_btn3.click(
-        lambda: load_sample_image(SAMPLE_CLOTHES["top"][2]),
+        lambda: load_sample_image(SAMPLE_CLOTHES["top"][2], outputs=[garment_top]),
         outputs=[garment_top]
     )
     top_btn4.click(
-        lambda: load_sample_image(SAMPLE_CLOTHES["top"][3]),
+        lambda: load_sample_image(SAMPLE_CLOTHES["top"][3], outputs=[garment_top]),
         outputs=[garment_top]
     )
     
     bottom_btn1.click(
-        lambda: load_sample_image(SAMPLE_CLOTHES["bottom"][0]),
+        lambda: load_sample_image(SAMPLE_CLOTHES["bottom"][0], outputs=[garment_down]),
         outputs=[garment_down]
     )
     bottom_btn2.click(
-        lambda: load_sample_image(SAMPLE_CLOTHES["bottom"][1]),
+        lambda: load_sample_image(SAMPLE_CLOTHES["bottom"][1], outputs=[garment_down]),
         outputs=[garment_down] 
     )
     bottom_btn3.click(
-        lambda: load_sample_image(SAMPLE_CLOTHES["bottom"][2]),
+        lambda: load_sample_image(SAMPLE_CLOTHES["bottom"][2], outputs=[garment_down]),
         outputs=[garment_down]
     )
     bottom_btn4.click(
-        lambda: load_sample_image(SAMPLE_CLOTHES["bottom"][3]),
+        lambda: load_sample_image(SAMPLE_CLOTHES["bottom"][3], outputs=[garment_down]),
         outputs=[garment_down]
     )