Fiqa commited on
Commit
2073533
·
verified ·
1 Parent(s): 4ea2dd5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -60,7 +60,7 @@ def generate_caption_and_image(image):
60
 
61
  pixel_values = feature_extractor(images=[img], return_tensors="pt").pixel_values.to(device)
62
  output_ids = model.generate(pixel_values, **gen_kwargs)
63
- caption = tokenizer.decode(output_ids[0], skip_special_tokens=True).strip()
64
 
65
 
66
 
 
60
 
61
  pixel_values = feature_extractor(images=[img], return_tensors="pt").pixel_values.to(device)
62
  output_ids = model.generate(pixel_values, **gen_kwargs)
63
+ caption1 = tokenizer.decode(output_ids[0], skip_special_tokens=True).strip()
64
 
65
 
66