Spaces:
Running
on
Zero
Running
on
Zero
ready for zeroGPU early tests
Browse files- gradio_app.py +2 -0
gradio_app.py
CHANGED
|
@@ -1,3 +1,4 @@
|
|
|
|
|
| 1 |
import os
|
| 2 |
import sys
|
| 3 |
import shutil
|
|
@@ -37,6 +38,7 @@ hf_hub_download(
|
|
| 37 |
local_dir = "./pretrained_weights"
|
| 38 |
)
|
| 39 |
|
|
|
|
| 40 |
def generate(control_sequence, ref_image):
|
| 41 |
control_image = control_sequence # "data_test/sample4.mp4"
|
| 42 |
ref_image = ref_image # "data_test/sample4.png"
|
|
|
|
| 1 |
+
import spaces
|
| 2 |
import os
|
| 3 |
import sys
|
| 4 |
import shutil
|
|
|
|
| 38 |
local_dir = "./pretrained_weights"
|
| 39 |
)
|
| 40 |
|
| 41 |
+
@spaces.GPU()
|
| 42 |
def generate(control_sequence, ref_image):
|
| 43 |
control_image = control_sequence # "data_test/sample4.mp4"
|
| 44 |
ref_image = ref_image # "data_test/sample4.png"
|