Spaces:
Runtime error
Runtime error
chats-bug
commited on
Commit
·
4bfc3de
1
Parent(s):
235543f
Trying to make the git large coco work
Browse files
app.py
CHANGED
@@ -69,7 +69,7 @@ def generate_caption(
|
|
69 |
|
70 |
generated_ids = model.generate(
|
71 |
pixel_values=pixel_values,
|
72 |
-
max_length=
|
73 |
)
|
74 |
|
75 |
if tokenizer is None:
|
@@ -139,8 +139,6 @@ def generate_captions(
|
|
139 |
if use_sample_image:
|
140 |
url = "http://images.cocodataset.org/val2017/000000039769.jpg"
|
141 |
image = Image.open(requests.get(url, stream=True).raw)
|
142 |
-
else:
|
143 |
-
image = Image.open(image)
|
144 |
|
145 |
# Generate captions for the image using the Blip base model
|
146 |
try:
|
|
|
69 |
|
70 |
generated_ids = model.generate(
|
71 |
pixel_values=pixel_values,
|
72 |
+
max_length=50,
|
73 |
)
|
74 |
|
75 |
if tokenizer is None:
|
|
|
139 |
if use_sample_image:
|
140 |
url = "http://images.cocodataset.org/val2017/000000039769.jpg"
|
141 |
image = Image.open(requests.get(url, stream=True).raw)
|
|
|
|
|
142 |
|
143 |
# Generate captions for the image using the Blip base model
|
144 |
try:
|