Requirements

utils/caption_utils.py

@@ -2,10 +2,8 @@ from transformers import BlipProcessor, BlipForConditionalGeneration, AutoTokeni
 import torch
 import requests
 from dotenv import load_dotenv
-from image_utils import UrlTest
 import os
 
-img = UrlTest()
 
 class ImageCaptioning:
     def __init__(self):
@@ -83,7 +81,6 @@ class ImageCaptioning:
         return [topic for topic in topics if topic and len(topic.split()) > 1]
 
     def combo_model(self, image, additional_text=None):
-        image = img.load_image(image)
         caption = self.generate_caption(image)
         caption = self.blip_processor.decode(caption[0], skip_special_tokens=True)
         topics = self.generate_topics(caption, additional_text)