Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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=
|
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 |
|