xinyuanc91 commited on
Commit
29d0b6e
Β·
1 Parent(s): f26051c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -8,6 +8,8 @@ from utils import mask_generation_before
8
  import os
9
  import cv2
10
 
 
 
11
  config_path = "./configs/sample_i2v.yaml"
12
  args = OmegaConf.load(config_path)
13
  device = "cuda" if torch.cuda.is_available() else "cpu"
 
8
  import os
9
  import cv2
10
 
11
+ os.system('wget https://huggingface.co/Vchitect/SEINE/blob/main/seine.pt -o pre-trained/')
12
+
13
  config_path = "./configs/sample_i2v.yaml"
14
  args = OmegaConf.load(config_path)
15
  device = "cuda" if torch.cuda.is_available() else "cpu"