Create app.py

app.py

@@ -0,0 +1,15 @@
+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'])