Update ClassPrompt.py

ClassPrompt.py

@@ -2,6 +2,9 @@ import random  # Import the random module
 from groq import Groq
 from openai import OpenAI
 import os
+import io
+import base64
+from huggingface_hub import InferenceApi
 class PromptClass:
     def __init__(self):
         self.huggingface_token = os.environ.get("HF_TOKEN")
@@ -303,4 +306,16 @@ Your output is only the caption itself, no comments or extra formatting. The cap
                 sentences = output.split(". ")
                 if len(sentences) > 1:
                     output = ". ".join(sentences[1:]).strip()
-            return output
+            return output
+    def img2text(self,image=None):
+        if image:
+            # Select the appropriate provider
+            inference = InferenceApi(repo_id="Salesforce/blip-image-captioning-base", token=self.huggingface_token)
+            # Đọc file hình ảnh
+            image_bytes = io.BytesIO()
+            image.save(image_bytes, format="JPEG")
+            image_data = image_bytes.getvalue()
+            image_base64 = base64.b64encode(image_data).decode("utf-8")
+            # Gửi yêu cầu API
+            response = inference(inputs={"image":image_base64})
+            return response[0]["generated_text"]