Update app.py

app.py

@@ -9,9 +9,6 @@ import subprocess
 from tqdm import tqdm
 import requests
 
-# Assuming spaces is a valid module
-import spaces
-
 def download_file(url, filename):
     response = requests.get(url, stream=True)
     total_size = int(response.headers.get('content-length', 0))
@@ -30,9 +27,8 @@ def download_file(url, filename):
 def setup_environment():
     if not os.path.exists("CCSR"):
         print("Cloning CCSR repository...")
-        subprocess.run(["git", "clone", "-b", "dev", "https://github.com/camenduru/CCSR.git"])
+        subprocess.run(["git", "clone", "-b", "dev", "https://github.com/camenduru/CCSR.git", "."])
     
-    os.chdir("CCSR")
     sys.path.append(os.getcwd())
     
     os.makedirs("weights", exist_ok=True)
@@ -47,11 +43,11 @@ def setup_environment():
 
 setup_environment()
 
-# Importing from the CCSR folder
-from CCSR.ldm.xformers_state import disable_xformers
-from CCSR.model.q_sampler import SpacedSampler
-from CCSR.model.ccsr_stage1 import ControlLDM
-from CCSR.utils.common import instantiate_from_config, load_state_dict
+# Importing from the current directory
+from ldm.xformers_state import disable_xformers
+from model.q_sampler import SpacedSampler
+from model.ccsr_stage1 import ControlLDM
+from utils.common import instantiate_from_config, load_state_dict
 
 config = OmegaConf.load("configs/model/ccsr_stage2.yaml")
 model = instantiate_from_config(config)
@@ -60,7 +56,6 @@ load_state_dict(model, ckpt, strict=True)
 model.freeze()
 model.to("cuda")
 
-@spaces.GPU  # Decorate the inference function with @spaces.GPU
 @torch.no_grad()
 def process(image, steps, t_max, t_min, color_fix_type):
     image = Image.open(image).convert("RGB")