Update CaptionGenerator.py

CaptionGenerator.py

@@ -39,7 +39,8 @@ class CaptionGenerator:
         :return: description - The description generated for the image.
         """
         try:
-            inputs = processor(images=image_data, return_tensors="pt")
+            prompt = " "
+            inputs = processor(images=image, text=prompt, return_tensors="pt")
             generated_ids = model.generate(**inputs, max_length=100)
             description = processor.batch_decode(generated_ids, skip_special_tokens=True)[0].strip()
             return description