Update input_video.py

2024-11-03 22:38:13 +08:00
parent 0904716cd0
commit 1faaad58f0
1 changed files with 2 additions and 2 deletions
@@ -151,7 +151,7 @@ class ShellAgentPluginInputVideo:
            num_bars = int(file_size / chunk_size)

            with open(video_path, "wb") as out_file:
-                for chunk in tqdm(
+                for chunk in tqdm.tqdm(
                    response.iter_content(chunk_size=chunk_size),
                    total=num_bars,
                    unit="KB",
@@ -175,4 +175,4 @@ NODE_CLASS_MAPPINGS = {
 NODE_DISPLAY_NAME_MAPPINGS = {
    # "ShellAgentPluginInputImage": "Input Image (ShellAgent Plugin)",
    "ShellAgentPluginInputVideo": "Input Video (ShellAgent Plugin)"
-}
+}