Suratsada's picture
Update app.py
4a995ec
raw
history blame
557 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)
if __name__ == "__main__":
main()