configure for HF spaces

app.py

@@ -5,6 +5,9 @@ from PIL import Image
 import io
 from omegaconf import OmegaConf
 
+import subprocess
+rc = subprocess.call("./setup.sh")
+
 import sys 
 import os
 sys.path.append(os.path.join(os.path.dirname(os.path.abspath(__file__)), 'lama'))

packages.txt

@@ -0,0 +1,3 @@
+git
+curl
+unzip

requirements.txt

@@ -19,3 +19,5 @@ kornia==0.5.0
 webdataset
 packaging
 wldhx.yadisk-direct
+torch
+torchvision

setup.sh

@@ -1,14 +1,18 @@
-conda create -n cleanup python=3.10 -y
-conda activate cleanup
-# conda install pytorch==1.9.0 torchvision==0.10.0 cudatoolkit=11.1 -c pytorch -c nvidia
-conda install pytorch torchvision -c pytorch -y
+#!/bin/bash
 
-pip install -r requirements.txt
+# Check if lama directory exists
+if [ ! -d "lama" ]; then
+    # Clone dependency repos
+    git clone https://github.com/advimman/lama.git
+else
+    echo "lama directory already exists. Skipping clone."
+fi
 
-
-# Clone dependency repos
-git clone https://github.com/advimman/lama.git
-
-# Download big-lama model
-curl -LJO https://huggingface.co/smartywu/big-lama/resolve/main/big-lama.zip
-unzip big-lama.zip
+# Check if big-lama.zip or big-lama directory exists
+if [ ! -f "big-lama.zip" ] && [ ! -d "big-lama" ]; then
+    # Download big-lama model
+    curl -LJO https://huggingface.co/smartywu/big-lama/resolve/main/big-lama.zip
+    unzip big-lama.zip
+else
+    echo "big-lama model already exists. Skipping download and extraction."
+fi

setup_local.sh

@@ -0,0 +1,14 @@
+conda create -n cleanup python=3.10 -y
+conda activate cleanup
+# conda install pytorch==1.9.0 torchvision==0.10.0 cudatoolkit=11.1 -c pytorch -c nvidia
+conda install pytorch torchvision -c pytorch -y
+
+pip install -r requirements.txt
+
+
+# Clone dependency repos
+git clone https://github.com/advimman/lama.git
+
+# Download big-lama model
+curl -LJO https://huggingface.co/smartywu/big-lama/resolve/main/big-lama.zip
+unzip big-lama.zip