make changes

app.py

@@ -13,6 +13,9 @@ from transformers import (
 from diffusers import DDPMScheduler,AutoencoderKL
 from typing import List
 
+import requests
+from io import BytesIO
+
 import torch
 import os
 from transformers import AutoTokenizer
@@ -249,6 +252,11 @@ def start_tryon(dict,garm_img,garment_des,is_checked,is_checked_crop,denoise_ste
     else:
         return images[0], mask_gray
     # return images[0], mask_gray
+# Function to handle image selection from the gallery
+def select_image(image_url):
+    response = requests.get(image_url)
+    image = Image.open(BytesIO(response.content))
+    return image
 
 garm_list = os.listdir(os.path.join(example_path,"cloth"))
 garm_list_path = [os.path.join(example_path,"cloth",garm) for garm in garm_list]
@@ -311,11 +319,8 @@ with image_blocks as demo:
 
 
     fetch_button.click(fn=fetch_products, inputs=prompt, outputs=[image_gallery, price_display, url_display])
-    def select_image(index, gallery_images):
-        if index is not None and index < len(gallery_images):
-            return gallery_images[index]
-        return None
-    image_gallery.select(fn=select_image, inputs=[gr.Slider(0, 9), image_gallery], outputs=garm_img)
+
+    image_gallery.select(fn=select_image, inputs=image_gallery, outputs=garm_img)
 
     try_button.click(fn=start_tryon, inputs=[imgs, garm_img, prompt, is_checked,is_checked_crop, denoise_steps, seed], outputs=[image_out,masked_img], api_name='tryon')