Spaces:
Sleeping
Sleeping
Wenzheng Chang
commited on
Commit
·
3e3a156
1
Parent(s):
da3b980
update app
Browse files
app.py
CHANGED
@@ -17,6 +17,7 @@ from diffusers import (
|
|
17 |
CogVideoXTransformer3DModel,
|
18 |
)
|
19 |
from transformers import AutoTokenizer, T5EncoderModel
|
|
|
20 |
|
21 |
|
22 |
rootutils.setup_root(__file__, indicator=".project-root", pythonpath=True)
|
@@ -423,6 +424,7 @@ def save_output_files(
|
|
423 |
return paths
|
424 |
|
425 |
|
|
|
426 |
def process_reconstruction(
|
427 |
video_file,
|
428 |
height,
|
@@ -545,6 +547,7 @@ def process_reconstruction(
|
|
545 |
return None, None, []
|
546 |
|
547 |
|
|
|
548 |
def process_prediction(
|
549 |
image_file,
|
550 |
height,
|
@@ -671,6 +674,7 @@ def process_prediction(
|
|
671 |
return None, None, []
|
672 |
|
673 |
|
|
|
674 |
def process_planning(
|
675 |
image_file,
|
676 |
goal_file,
|
|
|
17 |
CogVideoXTransformer3DModel,
|
18 |
)
|
19 |
from transformers import AutoTokenizer, T5EncoderModel
|
20 |
+
import huggingface_hub.spaces as spaces
|
21 |
|
22 |
|
23 |
rootutils.setup_root(__file__, indicator=".project-root", pythonpath=True)
|
|
|
424 |
return paths
|
425 |
|
426 |
|
427 |
+
@spaces.GPU(duration=240)
|
428 |
def process_reconstruction(
|
429 |
video_file,
|
430 |
height,
|
|
|
547 |
return None, None, []
|
548 |
|
549 |
|
550 |
+
@spaces.GPU(duration=240)
|
551 |
def process_prediction(
|
552 |
image_file,
|
553 |
height,
|
|
|
674 |
return None, None, []
|
675 |
|
676 |
|
677 |
+
@spaces.GPU(duration=240)
|
678 |
def process_planning(
|
679 |
image_file,
|
680 |
goal_file,
|