File size: 728 Bytes
3f375a2 337ae31 3f375a2 337ae31 3f375a2 beccfdf 3f375a2 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 |
import streamlit as st
from PIL import Image
import kosmos
st.title('KOSMOS2 Chat box')
uploaded_file = st.file_uploader("Pick an image...", type="jpg")
if uploaded_file is not None:
try:
image = Image.open(uploaded_file)
st.image(image, caption='Uploaded Image.', use_column_width=True)
st.write("")
prompt = st.text_input(f"Detect the main object in the image. The image is a")
if st.button('Classify'):
st.write(f"User: {prompt}")
with st.spinner('Processing...'):
label = kosmos.single_image_classification(image, prompt, 150)
st.write(f"Model: {label}")
except Exception as e:
st.error(f"An error occurred: {e}") |