Update app.py

app.py

@@ -1,11 +1,14 @@
 import gradio as gr
 import torch
-from transformers import AutoProcessor, Idefics3ForConditionalGeneration
+from transformers import Idefics3Processor, Idefics3ForConditionalGeneration
 from PIL import Image
 
 # Load model and processor
-model_id = "HuggingFaceM4/Idefics3-8B-Llama3" # Ensure this matches your model
-processor = AutoProcessor.from_pretrained(model_id, trust_remote_code=True)
+model_id = "HuggingFaceM4/Idefics3-8B-Llama3"
+
+# FIX: Import the specific processor class directly to bypass the AutoProcessor bug
+processor = Idefics3Processor.from_pretrained(model_id, trust_remote_code=True)
+
 model = Idefics3ForConditionalGeneration.from_pretrained(
     model_id, 
     trust_remote_code=True, 
@@ -16,22 +19,22 @@ def process_image(image):
     # Safety check for empty input
     if image is None:
         return "Please upload an image first."
-    
+        
     # Ensure image is PIL format
     if not isinstance(image, Image.Image):
         image = Image.fromarray(image)
-        
+            
     image = image.convert("RGB")
-    
+        
     # Prepare inputs
     messages = [{"role": "user", "content": [{"type": "image"}, {"type": "text", "text": "Describe this image."}]}]
     prompt = processor.apply_chat_template(messages, add_generation_prompt=True)
     inputs = processor(text=prompt, images=image, return_tensors="pt")
-    
+        
     # Generate
     generated_ids = model.generate(**inputs, max_new_tokens=500)
     result = processor.batch_decode(generated_ids, skip_special_tokens=True)
-    
+        
     return result[0]
 
 # UI Setup