Update app.py

app.py

@@ -53,6 +53,7 @@ image_processor = AutoImageProcessor.from_pretrained(MODEL_ID)
 processor = AutoProcessor.from_pretrained(MODEL_ID)
 model = LlavaForConditionalGeneration.from_pretrained(MODEL_ID, quantization_config=quantization_config, device_map="auto")
 # Pass the tokenizer, image processor explicitly to the pipeline
+pipe_image_to_text = pipeline("image-to-text", model=model, tokenizer=tokenizer_image_to_text, image_processor=image_processor, model_kwargs={"quantization_config": quantization_config})
 # Initialize the text generation pipeline
 
 pipe_text = pipeline(