From 82ef606c4b5baff5be223da162f323bd972930ab Mon Sep 17 00:00:00 2001 From: daxcay Date: Fri, 14 Jun 2024 18:41:26 +0530 Subject: [PATCH] Update DATASET_OpenAIChatImage.py --- classes/DATASET_OpenAIChatImage.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/classes/DATASET_OpenAIChatImage.py b/classes/DATASET_OpenAIChatImage.py index 353fb01..4cb17f4 100644 --- a/classes/DATASET_OpenAIChatImage.py +++ b/classes/DATASET_OpenAIChatImage.py @@ -38,7 +38,7 @@ class DATASET_OpenAIChatImage: def generate(self, image, image_detail, model, api_url, api_key, prompt, token_length): try: ai = OpenAI(api_key=api_key, base_url=api_url) - base64img = self.image_to_base64(image) + base64img = self.to_base64(image) if not api_key: return "OpenAI API key is required." request = [{"role": "system","content": "You are GPT-4"}]