| 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) |