Update text_to_video.py

text_to_video.py

@@ -1,5 +1,5 @@
 import torch
-
+from typing import Any
 from transformers.tools.base import Tool
 from transformers.utils import is_accelerate_available
 
@@ -17,9 +17,10 @@ TEXT_TO_VIDEO_DESCRIPTION = (
 class TextToVideoTool(Tool):
     default_checkpoint = "damo-vilab/text-to-video-ms-1.7b"
     description = TEXT_TO_VIDEO_DESCRIPTION
-    inputs = ['text']
-    outputs = ['video']
-
+    name = "video_generator"
+    inputs = {"prompt": str}
+    output_type = Any
+    
     def __init__(self, device=None, **hub_kwargs) -> None:
         if not is_accelerate_available():
             raise ImportError("Accelerate should be installed in order to use tools.")