chats-bug commited on
Commit
4bfc3de
·
1 Parent(s): 235543f

Trying to make the git large coco work

Browse files
Files changed (1) hide show
  1. app.py +1 -3
app.py CHANGED
@@ -69,7 +69,7 @@ def generate_caption(
69
 
70
  generated_ids = model.generate(
71
  pixel_values=pixel_values,
72
- max_length=64,
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: