Update vton.py

vton.py

@@ -40,10 +40,16 @@ def virtual_try_on(garm_img_url, human_img_url, garment_des):
         input=input_data
     )
     
+    # The model returns a list of URLs, get the first one
+    output_url = output[0]
+    
+    # Download the image from the URL
+    response = requests.get(output_url)
+    
     # Save the output image
     output_path = "output.jpg"
     with open(output_path, "wb") as file:
-        file.write(output.read())
+        file.write(response.content)
     
     print(f"Virtual try-on complete! Output saved to {output_path}")
     return output_path