Update app.py

conversion image

app.py

@@ -2,9 +2,22 @@ import gradio as gr
 from transformers import LlavaForConditionalGeneration, LlavaProcessor
 from PIL import Image
 import torch
+import io
 import warnings
 warnings.filterwarnings("ignore")
 
+
+def safe_convert_image(img):
+    if isinstance(img, Image.Image):
+        return img
+    elif isinstance(img, bytes):
+        return Image.open(io.BytesIO(img))
+    elif hasattr(img, "read"):  # Cas fichier Gradio
+        return Image.open(img)
+    else:
+        raise ValueError("Format d'image non pris en charge.")
+
+
 #Charger le modèle et le processeur
 model_id = "llava-hf/llava-1.5-7b-hf"
 #processor = LlavaProcessor.from_pretrained(model_id)
@@ -20,6 +33,8 @@ model = LlavaForConditionalGeneration.from_pretrained(
 def vqa_llava(image, question):
     if not isinstance(image, Image.Image):
         image = Image.open(image)
+        image = safe_convert_image(image)
+
 
     prompt = f"[INST] {question} [/INST]"
     inputs = processor(prompt, image, return_tensors="pt").to(model.device)