Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -6,10 +6,10 @@ caption = pipeline('image-to-text', model="ydshieh/vit-gpt2-coco-en")
|
|
6 |
|
7 |
uploaded_image=st.file_uploader("Upload an image", type=["ong", "jpg", "jpeg"])
|
8 |
|
9 |
-
if uploaded_image is not None
|
10 |
image = Image.open(uploaded_image)
|
11 |
st.Image(image, caption="Uploaded Image", use_column_width=True)
|
12 |
|
13 |
-
if st.button("Generate Captin")
|
14 |
captions = caption(image)
|
15 |
st.write(captions[0]['generated_text'])
|
|
|
6 |
|
7 |
uploaded_image=st.file_uploader("Upload an image", type=["ong", "jpg", "jpeg"])
|
8 |
|
9 |
+
if uploaded_image is not None:
|
10 |
image = Image.open(uploaded_image)
|
11 |
st.Image(image, caption="Uploaded Image", use_column_width=True)
|
12 |
|
13 |
+
if st.button("Generate Captin"):
|
14 |
captions = caption(image)
|
15 |
st.write(captions[0]['generated_text'])
|