added video_speed_tool

README.md

@@ -1,6 +1,6 @@
 ---
 title: Video Speed Tool
-emoji: 🏢
+emoji: 🏃
 colorFrom: yellow
 colorTo: blue
 sdk: gradio

app.py

@@ -0,0 +1,4 @@
+from transformers.tools.base import launch_gradio_demo
+from video_speed import VideoSpeedTool 
+
+launch_gradio_demo(VideoSpeedTool)

requirements.txt

@@ -0,0 +1,3 @@
+ffmpeg-python
+huggingface_hub
+transformers

tool_config.json

@@ -0,0 +1,6 @@
+{
+  "description": "This tool speeds up a video. Inputs are input_path as a string, output_path as a string, speed_factor (float) as a string. Output is the output_path.",
+  "name": "video_speed_tool",
+  "tool_class": "video_speed.VideoSpeedTool"
+}
+

video_speed.py

@@ -0,0 +1,19 @@
+import ffmpeg
+from transformers import Tool
+
+class VideoSpeedTool(Tool):  
+    name = "video_speed_tool"
+    description = """            
+    This tool speeds up a video.                     
+    Inputs are input_path as a string, output_path as a string, speed_factor (float) as a string.
+    Output is the output_path.
+    """                               
+    inputs = ["text", "text", "text"] 
+    outputs = ["text"]
+
+    def __call__(self, input_path: str, output_path: str, speed_factor: float):
+        stream = ffmpeg.input(input_path)
+        stream = ffmpeg.setpts(stream, "1/{}*PTS".format(float(speed_factor)))
+        stream = ffmpeg.output(stream, output_path)
+        ffmpeg.run(stream)
+        return output_path