SingleZombie
commited on
Commit
·
ac0890f
1
Parent(s):
593294b
update path
Browse files
app.py
CHANGED
@@ -36,6 +36,8 @@ from annotator.canny import CannyDetector
|
|
36 |
from annotator.midas import MidasDetector
|
37 |
|
38 |
import huggingface_hub
|
|
|
|
|
39 |
|
40 |
huggingface_hub.hf_hub_download('SingleZombie/FRESCO',
|
41 |
'boxer-punching-towards-camera.mp4',
|
@@ -52,15 +54,21 @@ huggingface_hub.hf_hub_download('SingleZombie/FRESCO',
|
|
52 |
|
53 |
huggingface_hub.hf_hub_download('PKUWilliamYang/Rerender',
|
54 |
'models/gmflow_sintel-0c07dcb3.pth',
|
55 |
-
local_dir='
|
56 |
|
57 |
huggingface_hub.hf_hub_download('PKUWilliamYang/Rerender',
|
58 |
'models/epoch_resnet.pth',
|
59 |
-
local_dir='
|
60 |
|
61 |
huggingface_hub.hf_hub_download('PKUWilliamYang/Rerender',
|
62 |
'models/ebsynth',
|
63 |
-
local_dir='src/ebsynth/deps/ebsynth/
|
|
|
|
|
|
|
|
|
|
|
|
|
64 |
|
65 |
device = 'cuda' if torch.cuda.is_available() else 'cpu'
|
66 |
|
|
|
36 |
from annotator.midas import MidasDetector
|
37 |
|
38 |
import huggingface_hub
|
39 |
+
import shutil
|
40 |
+
import os
|
41 |
|
42 |
huggingface_hub.hf_hub_download('SingleZombie/FRESCO',
|
43 |
'boxer-punching-towards-camera.mp4',
|
|
|
54 |
|
55 |
huggingface_hub.hf_hub_download('PKUWilliamYang/Rerender',
|
56 |
'models/gmflow_sintel-0c07dcb3.pth',
|
57 |
+
local_dir='./')
|
58 |
|
59 |
huggingface_hub.hf_hub_download('PKUWilliamYang/Rerender',
|
60 |
'models/epoch_resnet.pth',
|
61 |
+
local_dir='./')
|
62 |
|
63 |
huggingface_hub.hf_hub_download('PKUWilliamYang/Rerender',
|
64 |
'models/ebsynth',
|
65 |
+
local_dir='src/ebsynth/deps/ebsynth/')
|
66 |
+
|
67 |
+
shutil.move('models/gmflow_sintel-0c07dcb3.pth', 'model/gmflow_sintel-0c07dcb3.pth')
|
68 |
+
shutil.move('models/epoch_resnet.pth', 'model/epoch_resnet.pth')
|
69 |
+
shutil.move('src/ebsynth/deps/ebsynth/models/ebsynth', 'src/ebsynth/deps/ebsynth/bin/ebsynth')
|
70 |
+
|
71 |
+
shutil.rmtree('models')
|
72 |
|
73 |
device = 'cuda' if torch.cuda.is_available() else 'cpu'
|
74 |
|