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'])
|