merve HF staff commited on
Commit
8df48e3
1 Parent(s): b872a0b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -17,7 +17,7 @@ def infer_diagram(image, question):
17
  @spaces.GPU
18
  def infer_ocrvqa(image, question):
19
  model = PaliGemmaForConditionalGeneration.from_pretrained("google/paligemma-3b-ft-ocrvqa-896").to("cuda")
20
- processor = PaliGemmaProcessor.from_pretrained("google/paligemma-3b-ft-ocrvqa-896e")
21
 
22
  inputs = processor(images=image,text=question, return_tensors="pt").to("cuda")
23
 
 
17
  @spaces.GPU
18
  def infer_ocrvqa(image, question):
19
  model = PaliGemmaForConditionalGeneration.from_pretrained("google/paligemma-3b-ft-ocrvqa-896").to("cuda")
20
+ processor = PaliGemmaProcessor.from_pretrained("google/paligemma-3b-ft-ocrvqa-896")
21
 
22
  inputs = processor(images=image,text=question, return_tensors="pt").to("cuda")
23