Update models/image_text_transformation.py

models/image_text_transformation.py

@@ -25,14 +25,13 @@ class ImageTextTransformation:
         self.ref_image = None
     
     def init_models(self):
-        openai_key = os.getenv("OPENAI_KEY","")
         print(self.args)
         print('\033[1;34m' + "Welcome to the Image2Paragraph toolbox...".center(50, '-') + '\033[0m')
         print('\033[1;33m' + "Initializing models...".center(50, '-') + '\033[0m')
         print('\033[1;31m' + "This is time-consuming, please wait...".center(50, '-') + '\033[0m')
         self.image_caption_model = ImageCaptioning(device=self.args.image_caption_device, captioner_base_model=self.args.captioner_base_model)
         self.dense_caption_model = DenseCaptioning(device=self.args.dense_caption_device)
-        self.gpt_model = ImageToText(openai_key)
+        self.gpt_model = ImageToText()
         self.controlnet_model = TextToImage(device=self.args.contolnet_device)
         self.region_semantic_model = RegionSemantic(device=self.args.semantic_segment_device, image_caption_model=self.image_caption_model, region_classify_model=self.args.region_classify_model, sam_arch=self.args.sam_arch)
         print('\033[1;32m' + "Model initialization finished!".center(50, '-') + '\033[0m')