added video_compression_tool

README.md

@@ -1,6 +1,6 @@
 ---
 title: Video Compression Tool
-emoji: 🐠
+emoji: 🤏
 colorFrom: pink
 colorTo: blue
 sdk: gradio

app.py

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

requirements.txt

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

tool_config.json

@@ -0,0 +1,6 @@
+{
+  "description": "This tool compresses input video/gif to optimized video/gif. Inputs are input_path, output_path. Output is output_path.",
+  "name": "video_compression_tool",
+  "tool_class": "video_compression.VideoCompressionTool"
+}
+

video_compression.py

@@ -0,0 +1,21 @@
+import ffmpeg
+from transformers import Tool
+
+class VideoCompressionTool(Tool):
+    name = "video_compression_tool"
+    description = """
+    This tool compresses input video/gif to optimized video/gif. 
+    Inputs are input_path, output_path.
+    Output is output_path.
+    """
+    inputs = ["text", "text"]
+    outputs = ["text"]
+
+    def __call__(self, input_path: str, output_path: str):
+        (
+            ffmpeg.input(input_path)
+            .filter_("split", "[0:v] split [a][b];[a] palettegen [p];[b][p] paletteuse")
+            .output(output_path)
+            .run()
+        )
+        return output_path