Update app.py

app.py

@@ -58,8 +58,6 @@ def load_models():
         raise
 
 def generate_tryon(person_img, cloth_img):
-    """CRITICAL: Removed progress parameter - causes API issues"""
-    
     print("="*50, file=sys.stderr)
     print(f"Received - Person: {type(person_img)}, Cloth: {type(cloth_img)}", file=sys.stderr)
     
@@ -67,13 +65,8 @@ def generate_tryon(person_img, cloth_img):
         raise gr.Error("Both images required!")
     
     try:
-        # Convert filepaths to PIL
-        if isinstance(person_img, str):
-            person_img = Image.open(person_img).convert('RGB')
-        if isinstance(cloth_img, str):
-            cloth_img = Image.open(cloth_img).convert('RGB')
-        
-        print("Images converted to PIL", file=sys.stderr)
+        # person_img and cloth_img are already PIL Images, no conversion needed
+        print("Images received as PIL", file=sys.stderr)
         
         load_models()
         
@@ -108,10 +101,10 @@ def generate_tryon(person_img, cloth_img):
 demo = gr.Interface(
     fn=generate_tryon,
     inputs=[
-        gr.Image(label="Person Image", type="filepath"),
-        gr.Image(label="Garment Image", type="filepath")
+        gr.Image(label="Person Image", type="pil"),  # Changed from "filepath" to "pil"
+        gr.Image(label="Garment Image", type="pil")  # Changed from "filepath" to "pil"
     ],
-    outputs=gr.Image(label="Result", type="filepath"),
+    outputs=gr.Image(label="Result", type="pil"),  # Changed from "filepath" to "pil"
     title="Try-Space Virtual Try-On",
     description="Upload person and garment images. Processing takes 2-3 minutes on GPU T4.",
     api_name="generate_tryon",