SingleZombie commited on
Commit
ac0890f
1 Parent(s): 593294b

update path

Browse files
Files changed (1) hide show
  1. app.py +11 -3
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='model')
56
 
57
  huggingface_hub.hf_hub_download('PKUWilliamYang/Rerender',
58
  'models/epoch_resnet.pth',
59
- local_dir='model')
60
 
61
  huggingface_hub.hf_hub_download('PKUWilliamYang/Rerender',
62
  'models/ebsynth',
63
- local_dir='src/ebsynth/deps/ebsynth/bin')
 
 
 
 
 
 
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