Update app.py

app.py

@@ -1,4 +1,5 @@
 import os
+import sys
 import torch
 import gradio as gr
 import spaces
@@ -7,6 +8,7 @@ import numpy as np
 from omegaconf import OmegaConf
 import requests
 from tqdm import tqdm
+import subprocess
 
 def download_file(url, filename):
     response = requests.get(url, stream=True)
@@ -24,6 +26,13 @@ def download_file(url, filename):
             progress_bar.update(size)
 
 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"])
+    
+    os.chdir("CCSR")
+    sys.path.append(os.getcwd())
+    
     os.makedirs("weights", exist_ok=True)
     if not os.path.exists("weights/real-world_ccsr.ckpt"):
         print("Downloading model checkpoint...")