Spaces:
Sleeping
Sleeping
Commit
·
e6ba862
1
Parent(s):
864fa38
changes
Browse files
app.py
CHANGED
|
@@ -402,8 +402,8 @@ def predict_and_score_once(image: Image.Image, line_id: int = 1, topk: int = TOP
|
|
| 402 |
try:
|
| 403 |
out = model.generate(
|
| 404 |
pixel_values,
|
| 405 |
-
max_length=
|
| 406 |
-
num_beams=
|
| 407 |
do_sample=False,
|
| 408 |
return_dict_in_generate=True,
|
| 409 |
output_scores=True,
|
|
|
|
| 402 |
try:
|
| 403 |
out = model.generate(
|
| 404 |
pixel_values,
|
| 405 |
+
max_length=100,
|
| 406 |
+
num_beams=1,
|
| 407 |
do_sample=False,
|
| 408 |
return_dict_in_generate=True,
|
| 409 |
output_scores=True,
|