app.py

# Import convention
>>> import streamlit as st
from transformers import pipeline
from PIL import Image

# Load the image-to-text pipeline
image_read = pipeline("image-to-text", model="Salesforce/blip-image-captioning-base")

def read(image):
    results = image_read(image)
    return results

# Streamlit UI
st.title("Storytelling from images for 3-10 year-old kids")

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

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

    # Read image
    image_content = read(image)

    # Display results
    st.subheader("Image Content:")
    st.write(f"Image Content: {image_content[0]}")