Spaces:
Runtime error
Runtime error
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 |
|