import gradio as gr
import os

HF_TOKEN = os.getenv('HF_TOKEN')

hf_saver = gr.HuggingFaceDatasetSaver(HF_TOKEN, "test-image-classifier-dataset")

gr.Interface.load("huggingface/google/vit-base-patch16-224", flagging_callback=hf_saver).launch()