Update image_captioner.py

2025-05-17 02:28:12 +05:30
parent 0750d19ced
commit 580981fefc
1 changed files with 5 additions and 1 deletions
@@ -1,6 +1,10 @@
 import torch
 import os
 from PIL import Image
+import os
+import time
+from transformers import pipeline
+
 from transformers import (
    AutoModelForImageTextToText,
    AutoProcessor
@@ -113,5 +117,5 @@ NODE_CLASS_MAPPINGS = {
 }

 NODE_DISPLAY_NAME_MAPPINGS = {
-    "ImageCaptioner": "ImageCaptioner helps to generate captions for images in a folder and rename them sequentially.",
+    "ImageCaptioner": "Image Captioner",
 }