text-to-video

Running

lysandre HF staff commited on May 3, 2023

Commit

d170a8b

•

1 Parent(s): 280e8b0

IO

Files changed (1) hide show

text_to_video.py CHANGED Viewed

@@ -17,6 +17,8 @@ TEXT_TO_VIDEO_DESCRIPTION = (
 class TextToVideoTool(Tool):
     default_checkpoint = "damo-vilab/text-to-video-ms-1.7b"
     description = TEXT_TO_VIDEO_DESCRIPTION
     def __init__(self, device=None, **hub_kwargs) -> None:
         if not is_accelerate_available():

 class TextToVideoTool(Tool):
     default_checkpoint = "damo-vilab/text-to-video-ms-1.7b"
     description = TEXT_TO_VIDEO_DESCRIPTION
+    inputs = ['text']
+    outputs = ['video']
     def __init__(self, device=None, **hub_kwargs) -> None:
         if not is_accelerate_available():