resolve build errors

app.py

@@ -11,6 +11,8 @@ import spaces
 
 import gradio as gr
 
+print("Imports done, downloading the model...")
+
 # Download checkpoints
 snapshot_download(repo_id="franciszzj/Leffa", local_dir="./ckpts")
 
@@ -36,6 +38,8 @@ pt_model = LeffaModel(
 )
 pt_inference = LeffaInference(model=pt_model)
 
+print("Model downloaded, ready to serve!")
+
 @spaces.GPU
 def leffa_predict(src_image_path, ref_image_path, control_type):
     assert control_type in [
@@ -89,10 +93,12 @@ def leffa_predict_vt(src_image_path, ref_image_path):
     return leffa_predict(src_image_path, ref_image_path, "virtual_tryon")
 
 @spaces.GPU
-def leffa_predict_vt_image_url(src_image_path, ref_image_path):
-    src_image = fetch_image_from_url(src_image_path)
+def leffa_predict_vt_image_url(person_url, garment_url):
+    if not person_url or not garment_url:
+        return None
+    src_image = fetch_image_from_url(person_url)
     print("fetched person image")
-    ref_image = fetch_image_from_url(ref_image_path)
+    ref_image = fetch_image_from_url(garment_url)
     print("fetched garment image")
 
     return leffa_predict(src_image, ref_image, "virtual_tryon")
@@ -289,7 +295,7 @@ def create_demo():
                 )
         
         generate_btn.click(
-            fn=virtual_tryon_from_urls,
+            fn=leffa_predict_vt_image_url,
             inputs=[person_url, garment_url],
             outputs=output_image
         )

requirements.txt

@@ -29,4 +29,4 @@ torchvision
 tqdm
 transformers
 triton
-cloudpickle
+cloudpickle