Spaces:
Runtime error
Runtime error
Upload app.py
Browse files
app.py
CHANGED
@@ -16,11 +16,7 @@ def generate_caption(pixel_values):
|
|
16 |
with torch.no_grad():
|
17 |
generated_ids = model.generate(
|
18 |
pixel_values=pixel_values,
|
19 |
-
max_length=
|
20 |
-
num_beams=4,
|
21 |
-
early_stopping=True,
|
22 |
-
no_repeat_ngram_size=2
|
23 |
-
)
|
24 |
return processor.batch_decode(generated_ids, skip_special_tokens=True)[0]
|
25 |
|
26 |
def translate_to_portuguese(text):
|
|
|
16 |
with torch.no_grad():
|
17 |
generated_ids = model.generate(
|
18 |
pixel_values=pixel_values,
|
19 |
+
max_length=500)
|
|
|
|
|
|
|
|
|
20 |
return processor.batch_decode(generated_ids, skip_special_tokens=True)[0]
|
21 |
|
22 |
def translate_to_portuguese(text):
|