Update app.py

app.py

@@ -16,8 +16,8 @@ def infer_diagram(image, question):
 
 @spaces.GPU
 def infer_ocrvqa(image, question):
-    model = Pix2StructForConditionalGeneration.from_pretrained("google/paligemma-3b-ft-ocrvqa-896").to("cuda")
-    processor = Pix2StructProcessor.from_pretrained("google/paligemma-3b-ft-ocrvqa-896e")
+    model = PaliGemmaForConditionalGeneration.from_pretrained("google/paligemma-3b-ft-ocrvqa-896").to("cuda")
+    processor = PaliGemmaProcessor.from_pretrained("google/paligemma-3b-ft-ocrvqa-896e")
 
     inputs = processor(images=image,text=question, return_tensors="pt").to("cuda")