Wenzheng Chang commited on
Commit
3e3a156
·
1 Parent(s): da3b980

update app

Browse files
Files changed (1) hide show
  1. app.py +4 -0
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,