update

app.py

@@ -3,6 +3,7 @@ import os
 import time
 import sys
 import torch
+import zipfile
 import gradio as gr
 import u2net_load
 import u2net_run
@@ -59,8 +60,11 @@ if not os.path.exists("saved_models/u2net/u2net.pth"):
 # Get model checkpoints
 if not os.path.exists("./checkpoints/decavtonfifapretrain/"):
     os.system("wget -O ./checkpoints/decavtonfifapretrain.zip https://github.com/hasibzunair/vton-demo/releases/download/v1.0/decavtonfifapretrain.zip")
-    os.system("sudo apt-get install zip unzip")
-    os.system("unzip ./checkpoints/decavtonfifapretrain.zip -d ./checkpoints/")
+    # Works locally
+    #os.system("sudo apt-get install zip unzip")
+    #os.system("unzip ./checkpoints/decavtonfifapretrain.zip -d ./checkpoints/")
+    with zipfile.ZipFile('./checkpoints/decavtonfifapretrain.zip', 'r') as zip_ref:
+        zip_ref.extractall('./checkpoints/')
 
 print("########################Setup done!########################")