Suratsada's picture
Create new file
ecc85fd
raw
history blame
498 Bytes
import streamlit as st
from transformers import pipeline
classifier = pipeline("image-classification",
model="microsoft/cvt-13")
def main():
st.title("image-classification")
with st.form("image-classification"):
text = st.text_area('enter link image:')
# clicked==True only when the button is clicked
clicked = st.form_submit_button("Submit")
if clicked:
results = classifier([text])
st.(results)