Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
#2
by
euginfrancis
- opened
app.py
CHANGED
@@ -85,7 +85,7 @@ def caption_image(image, custom_prompt=None):
|
|
85 |
inputs=input_ids,
|
86 |
attention_mask=attention_mask,
|
87 |
images=px,
|
88 |
-
max_new_tokens=
|
89 |
do_sample=False, # Deterministic generation
|
90 |
temperature=1.0,
|
91 |
)
|
|
|
85 |
inputs=input_ids,
|
86 |
attention_mask=attention_mask,
|
87 |
images=px,
|
88 |
+
max_new_tokens=1000,
|
89 |
do_sample=False, # Deterministic generation
|
90 |
temperature=1.0,
|
91 |
)
|