keplersj commited on
Commit
d253a79
·
1 Parent(s): 3558088

conditional

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