Spaces:
Running
Running
Update app.py
Browse files
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 |
-
|
10 |
-
st.image(
|
11 |
if st.button("Generate caption"):
|
12 |
-
captions=caption(
|
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'])
|