Spaces:

merve
/

paligemma-doc

Running on Zero

merve HF staff commited on May 16

Commit

8df48e3

•

1 Parent(s): b872a0b

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -17,7 +17,7 @@ def infer_diagram(image, question):
 @spaces.GPU
 def infer_ocrvqa(image, question):
     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")

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