File size: 245 Bytes
0420f0c
63399bb
0420f0c
63399bb
0420f0c
 
 
 
1
2
3
4
5
6
7
8
9
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()