m-ric HF staff commited on
Commit
d90d4fc
1 Parent(s): 1f99294

Update text_to_video.py

Browse files
Files changed (1) hide show
  1. text_to_video.py +3 -3
text_to_video.py CHANGED
@@ -8,8 +8,8 @@ if is_accelerate_available():
8
  from accelerate import PartialState
9
 
10
  TEXT_TO_VIDEO_DESCRIPTION = (
11
- "This is a tool that creates a video according to a text description. It takes an input named `prompt` which "
12
- "contains the image description, as well as an optional input `seconds` which will be the duration of the video. "
13
  "The default is of two seconds. The tool outputs a video object."
14
  )
15
 
@@ -18,7 +18,7 @@ class TextToVideoTool(Tool):
18
  default_checkpoint = "damo-vilab/text-to-video-ms-1.7b"
19
  description = TEXT_TO_VIDEO_DESCRIPTION
20
  name = "video_generator"
21
- inputs = {"prompt": str}
22
  output_type = Any
23
 
24
  def __init__(self, device=None, **hub_kwargs) -> None:
 
8
  from accelerate import PartialState
9
 
10
  TEXT_TO_VIDEO_DESCRIPTION = (
11
+ "This is a tool that creates a video according to a text description."
12
+ "It takes an optional input `seconds` which will be the duration of the video."
13
  "The default is of two seconds. The tool outputs a video object."
14
  )
15
 
 
18
  default_checkpoint = "damo-vilab/text-to-video-ms-1.7b"
19
  description = TEXT_TO_VIDEO_DESCRIPTION
20
  name = "video_generator"
21
+ inputs = {"prompt": {"type": str, "description": "contains the image description"}}
22
  output_type = Any
23
 
24
  def __init__(self, device=None, **hub_kwargs) -> None: