Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -122,10 +122,10 @@ def generate_description(image):
|
|
| 122 |
output = model_gptneo.generate(
|
| 123 |
input_ids=input_ids,
|
| 124 |
attention_mask=attention_mask,
|
| 125 |
-
max_length=
|
| 126 |
temperature=0.7,
|
| 127 |
top_p=0.9,
|
| 128 |
-
repetition_penalty=1.
|
| 129 |
do_sample=True,
|
| 130 |
pad_token_id=tokenizer.pad_token_id
|
| 131 |
)
|
|
|
|
| 122 |
output = model_gptneo.generate(
|
| 123 |
input_ids=input_ids,
|
| 124 |
attention_mask=attention_mask,
|
| 125 |
+
max_length=250,
|
| 126 |
temperature=0.7,
|
| 127 |
top_p=0.9,
|
| 128 |
+
repetition_penalty=1.5,
|
| 129 |
do_sample=True,
|
| 130 |
pad_token_id=tokenizer.pad_token_id
|
| 131 |
)
|