Scharbhen commited on
Commit
485c677
·
verified ·
1 Parent(s): ead20dc

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -21,7 +21,7 @@ def infer_ocrvqa(image, question):
21
 
22
  inputs = processor(images=image,text=question, return_tensors="pt").to("cuda")
23
 
24
- predictions = model.generate(**inputs, max_new_tokens=500)
25
 
26
  return processor.decode(predictions[0], skip_special_tokens=True)[len(question):].lstrip("\n")
27
 
 
21
 
22
  inputs = processor(images=image,text=question, return_tensors="pt").to("cuda")
23
 
24
+ predictions = model.generate(**inputs, max_new_tokens=200)
25
 
26
  return processor.decode(predictions[0], skip_special_tokens=True)[len(question):].lstrip("\n")
27