Update app.py

app.py

@@ -3,7 +3,7 @@ import torch
 import os
 import tempfile
 from huggingface_hub import login
-from transformers import AutoProcessor, AutoModelForVisualQuestionAnswering, infer_device, PaliGemmaForConditionalGeneration
+from transformers import AutoProcessor, AutoModelForVisualQuestionAnswering, infer_device, pipeline
 from accelerate import Accelerator
 
 # Set the device
@@ -37,12 +37,9 @@ def process_image(image, prompt):
 
 # MODEL 2: PaliGemma ---
 
-model2_id = "merve/paligemma_vqav2"
+
 processor2 = AutoProcessor.from_pretrained("google/paligemma-3b-pt-224")
-model2 = PaliGemmaForConditionalGeneration.from_pretrained(
-    model2_id, 
-    torch_dtype=torch.bfloat16
-).to(device)
+model2 = pipeline("image-to-text", model="merve/paligemma_vqav2")
 
 
 # Define inference function for Model 2