File size: 1,408 Bytes
216efed
a968ae9
216efed
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
cd94435
 
216efed
 
 
 
 
 
a968ae9
 
216efed
 
a968ae9
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
import os
import re
import sys
import git
import subprocess
from huggingface_hub import hf_hub_download

REPO_URL = "https://github.com/facebookresearch/videoseal.git"
REPO_BRANCH = '5897ac50b5b0f5c806f42d2f7d1ef208a0780a28'
LOCAL_PATH = "./videoseal"

def install_src():
    if not os.path.exists(LOCAL_PATH):
        print(f"Cloning repository from {REPO_URL}...")
        repo = git.Repo.clone_from(REPO_URL, LOCAL_PATH)
        repo.git.checkout(REPO_BRANCH)
    else:
        print(f"Repository already exists at {LOCAL_PATH}")

    requirements_path = os.path.join(LOCAL_PATH, "requirements.txt")
    if os.path.exists(requirements_path):
        print("Installing requirements...")
        # Lock versions by attaching a custom requirements.txt.
        subprocess.check_call(["pip", "install", "-r", "./requirements.txt", "-r", requirements_path])
    else:
        print("No requirements.txt found.")

# clone repo
install_src()

# set app path form import
os.environ["APP_PATH"] = LOCAL_PATH

# run gradio
# from gradio.cli import cli
# if __name__ == '__main__':
#     sys.argv[0] = re.sub(r'(-script\.pyw|\.exe)?$', '', sys.argv[0]) ##os.path.join(os.path.dirname(sys.argv[0]), "gradio_run.py")
#     sys.argv.append(os.path.join(os.path.dirname(sys.argv[0]), "gradio_app.py"))
#     sys.exit(cli())

subprocess.check_call(["python", os.path.join(os.path.dirname(sys.argv[0]), "gradio_app.py")])