From 580981fefcedd542b9e8754584b603228005883d Mon Sep 17 00:00:00 2001
From: AIExplorer <abuzar332@gmail.com>
Date: Sat, 17 May 2025 02:28:12 +0530
Subject: [PATCH] Update image_captioner.py

---
 image_captioner.py | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/image_captioner.py b/image_captioner.py
index b5673c0..23332ec 100644
--- a/image_captioner.py
+++ b/image_captioner.py
@@ -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",
 }