from transformers import pipeline
import gradio as gr

pipe = pipeline("image-classification", model="google/vit-base-patch16-224")

gr.Interface.from_pipeline(pipe).launch()

# import streamlit as st
# from transformers import pipeline

# transcriber = pipeline(task="sentiment-analysis")
# text = st.text_input('Label', 'enter some text!')

# if text:
#     out = transcriber(text)
#     st.json(out)

# uploaded_file = st.file_uploader("Choose a CSV file", accept_multiple_files=True)
# bytes_data = uploaded_file.read()
# st.write("filename:", uploaded_file.name)
# st.write(bytes_data)
# st.image(uploaded_file)