import streamlit as st
from transformers import pipeline
from PIL import Image

caption = pipeline('image-to-text', model="ydshieh/vit-gpt2-coco-en")

uploaded_image=st.file_uploader("Upload an image", type=["ong", "jpg", "jpeg"])

if uploaded_image is not None:
   image = Image.open(uploaded_image)
   st.image(image, caption="Uploaded Image", use_column_width=True)

if st.button("Generate Captin"):
   captions = caption(image)
   st.write(captions[0]['generated_text'])