Update app.py

app.py

@@ -8,7 +8,6 @@ from omegaconf import OmegaConf
 import subprocess
 from tqdm import tqdm
 import requests
-import shutil
 
 def download_file(url, filename):
     response = requests.get(url, stream=True)
@@ -26,28 +25,11 @@ def download_file(url, filename):
             progress_bar.update(size)
 
 def setup_environment():
-    if not os.path.exists("CCSR_temp"):
+    if not os.path.exists("CCSR"):
         print("Cloning CCSR repository...")
-        subprocess.run(["git", "clone", "-b", "dev", "https://github.com/camenduru/CCSR.git", "CCSR_temp"])
-    else:
-        print("CCSR repository already cloned. Skipping clone step.")
-    
-    # Move necessary files/directories from CCSR_temp to current directory
-    for item in os.listdir("CCSR_temp"):
-        s = os.path.join("CCSR_temp", item)
-        d = os.path.join(os.getcwd(), item)
-        if os.path.isdir(s):
-            if os.path.exists(d):
-                shutil.rmtree(d)
-            shutil.move(s, d)
-        else:
-            if os.path.exists(d):
-                os.remove(d)
-            shutil.move(s, d)
-    
-    # Clean up the temporary directory
-    shutil.rmtree("CCSR_temp")
+        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)
@@ -62,7 +44,7 @@ def setup_environment():
 
 setup_environment()
 
-# Importing from the current directory
+# Importing from the CCSR folder
 from ldm.xformers_state import disable_xformers
 from model.q_sampler import SpacedSampler
 from model.ccsr_stage1 import ControlLDM