Update vton.py

vton.py

@@ -3,32 +3,27 @@ import requests
 import os
 import base64
 
-def virtual_try_on(garm_img_url, human_img_url, garment_des):
+def virtual_try_on(garm_img_data, human_img_data, garment_des):
     """
     Perform virtual try-on using the IDM-VTON model on Replicate.
     
     Args:
-        garm_img_url (str): Path or URL of the garment image
-        human_img_url (str): Path or URL of the human image
+        garm_img_data (bytes): Binary data of the garment image
+        human_img_data (bytes): Binary data of the human image
         garment_des (str): Description of the garment
         
     Returns:
         str: Path to the saved output image
     """
-    # Convert local files to data URLs
-    def file_to_data_url(file_path):
-        if not os.path.exists(file_path):
-            raise ValueError(f"File not found: {file_path}")
-            
-        with open(file_path, 'rb') as f:
-            img_data = f.read()
+    # Convert binary data to data URLs
+    def bytes_to_data_url(img_data):
         encoded = base64.b64encode(img_data).decode('utf-8')
         return f"data:image/png;base64,{encoded}"
     
     # Prepare input for the model
     input_data = {
-        "garm_img": file_to_data_url(garm_img_url),
-        "human_img": file_to_data_url(human_img_url),
+        "garm_img": bytes_to_data_url(garm_img_data),
+        "human_img": bytes_to_data_url(human_img_data),
         "garment_des": garment_des
     }