Update vton.py

vton.py

@@ -35,10 +35,18 @@ def virtual_try_on(garm_img_data, human_img_data, garment_des):
     )
     
     # The model returns a list of URLs, get the first one
+    if not output or not isinstance(output, list):
+        raise ValueError("Invalid response from the model")
+        
     output_url = output[0]
     
+    # Validate URL scheme
+    if not output_url.startswith(('http://', 'https://')):
+        output_url = 'https://' + output_url
+    
     # Download the image from the URL
     response = requests.get(output_url)
+    response.raise_for_status()  # Raise an exception for bad status codes
     
     # Save the output image
     output_path = "output.jpg"