Cheselle commited on
Commit
831b696
·
verified ·
1 Parent(s): d1501ae

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -9,7 +9,7 @@ model = BlipForConditionalGeneration.from_pretrained("Salesforce/blip-image-capt
9
  file_name = st.file_uploader("Upload")
10
 
11
  if file_name is not None:
12
- inputs = processor(Image.open(file_name), return_tensors="pt")
13
 
14
  out = model.generate(**inputs)
15
- st.write(processor.decode(out[0], skip_special_tokens=True))
 
9
  file_name = st.file_uploader("Upload")
10
 
11
  if file_name is not None:
12
+ inputs = processor(Image.open(file_name).convert('RGB'), return_tensors="pt")
13
 
14
  out = model.generate(**inputs)
15
+ print(processor.decode(out[0], skip_special_tokens=True))