vmarellano commited on
Commit
c9096f7
·
verified ·
1 Parent(s): abdf527

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -14,8 +14,8 @@ raw_image = Image.open(requests.get(img_url, stream=True).raw).convert('RGB')
14
  text = "a photography of"
15
  inputs = processor(raw_image, text, return_tensors="pt")
16
 
17
- #outputs = model.generate(inputs, clean_up_tokenization_spaces=True)
18
- out = model.generate(**inputs, clean_up_tokenization_spaces=True, max_length = 2400)
19
  print(processor.decode(out[0], skip_special_tokens=True))
20
 
21
  # unconditional image captioning
 
14
  text = "a photography of"
15
  inputs = processor(raw_image, text, return_tensors="pt")
16
 
17
+ # out = model.generate(**inputs, clean_up_tokenization_spaces=True, max_length = 2400)
18
+ out = model.generate(**inputs, clean_up_tokenization_spaces=True)
19
  print(processor.decode(out[0], skip_special_tokens=True))
20
 
21
  # unconditional image captioning