initial commit

README.md

@@ -1,7 +1,7 @@
 ---
 title: Video Stabilizer
-emoji: 😻
-colorFrom: red
+emoji: 🐎
+colorFrom: green
 colorTo: indigo
 sdk: gradio
 sdk_version: 3.43.2

app.py

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

requirements.txt

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

tool_config.json

@@ -0,0 +1,6 @@
+{
+  "description": "This tool stabilizes a video. Inputs are input_path, output_path, smoothing, zoom. Output is the output_path.",
+  "name": "video_stabilization_tool",
+  "tool_class": "video_stabilizer.VideoStabilizationTool"
+}
+

video_stabilizer.py

@@ -0,0 +1,38 @@
+import ffmpeg
+from transformers import Tool
+
+class VideoStabilizationTool(Tool):
+    name = "video_stabilization_tool"
+    description = """
+    This tool stabilizes a video.
+    Inputs are input_path, output_path, smoothing, zoom.
+    """
+    inputs = ["text", "text", "integer", "integer", "integer"]
+    outputs = ["text"]
+
+    def __call__(
+        self,
+        input_path: str,
+        output_path: str,
+        smoothing: int = 10,
+        zoom: int = 0,
+        shakiness: int = 5,
+    ):
+        (
+            ffmpeg.input(input_path)
+            .output("null", vf="vidstabdetect=shakiness={}".format(shakiness), f="null")
+            .overwrite_output()
+            .run(capture_stdout=True, capture_stderr=True)
+        )
+        (
+            ffmpeg.input(input_path)
+            .output(
+                output_path,
+                vf="vidstabtransform=smoothing={}:zoom={}:input={}".format(
+                    smoothing, zoom, "transforms.trf"
+                ),
+            )
+            .overwrite_output()
+            .run()
+        )
+        return output_path