import streamlit as st
from transformers import pipeline
from datasets import load_dataset

pipeline = pipeline(task="image-classification", model="puqi/1")
data_files = {"train": "train_input_small.npy", "val": "val_input_small.npy"}
dataset = load_dataset("puqi/test ", data_files=data_files)


st.table(dataset)