Update app.py

app.py

@@ -5,7 +5,11 @@ import gradio as gr
 from transformers import LlavaNextProcessor, LlavaNextForConditionalGeneration
 
 model_path = "ibm-granite/granite-vision-3.1-2b-preview"
-processor = LlavaNextProcessor.from_pretrained(model_path, use_fast=True)
+processor = LlavaNextProcessor.from_pretrained(
+    model_path, 
+    use_fast=True,
+    image_processor_type = 'llava_next'
+)
 model = LlavaNextForConditionalGeneration.from_pretrained(model_path, torch_dtype="auto", device_map="auto")
 
 @spaces.GPU()