An-j96 commited on
Commit
27b4fd2
1 Parent(s): c171e94

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -6,8 +6,8 @@ from PIL import Image
6
  caption=pipeline('image-to-text',model='ydshieh/vit-gpt2-coco-en')
7
  uploaded_image=st.file_uploader("upload an image", type=["png","jpg","jpeg"])
8
  if uploaded_image is not None:
9
- image=image.open(uploaded_image)
10
- st.image(image, caption="uploaded_image", use_column_width=True)
11
  if st.button("Generate caption"):
12
- captions=caption(image)
13
  st.write(captions[0],['generated text'])
 
6
  caption=pipeline('image-to-text',model='ydshieh/vit-gpt2-coco-en')
7
  uploaded_image=st.file_uploader("upload an image", type=["png","jpg","jpeg"])
8
  if uploaded_image is not None:
9
+ Image=Image.open(uploaded_image)
10
+ st.image(Image, caption="uploaded_image", use_column_width=True)
11
  if st.button("Generate caption"):
12
+ captions=caption(Image)
13
  st.write(captions[0],['generated text'])