added image_directory_to_video_tool

README.md

@@ -1,6 +1,6 @@
 ---
-title: Image Directory To Video
-emoji: 📈
+title: Image Directory To Video Tool
+emoji: 🖼
 colorFrom: red
 colorTo: pink
 sdk: gradio

app.py

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

image_directory_to_video.py

@@ -0,0 +1,32 @@
+import json
+import ffmpeg
+from transformers import Tool
+
+class ImageDirectoryToVideoTool(Tool):
+    name = "image_directory_video_tool"
+    description = """
+    This tool creates video
+    from a directory of images. Inputs
+    are input_path and output_path.        
+    Output is the output_path.
+    """
+    inputs = ["text", "text"]
+    outputs = ["text"]
+
+    def __call__(
+        self,
+        input_path: str,
+        output_path: str,
+        framerate: int = 25,
+        extension: str = "jpg",
+    ):
+        (
+            ffmpeg.input(
+                input_path.rstrip("/") + "/*." + extension.lstrip("."),
+                pattern_type="glob",
+                framerate=framerate,
+            )
+            .output(output_path)
+            .run()
+        )
+        return output_path

requirements.txt

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

tool_config.json

@@ -0,0 +1,6 @@
+{
+  "description": "This tool creates video from a directory of images. Inputs are input_path and output_path. Output is the output_path.",
+  "name": "image_directory_video_tool",
+  "tool_class": "image_directory_to_video.ImageDirectoryToVideoTool"
+}
+