File size: 454 Bytes
d9f024f 63399bb adf1c79 40c7d40 d9f024f 97f1cf3 d9f024f 0420f0c 63399bb 0420f0c |
1 2 3 4 5 6 7 8 9 10 11 12 |
#### ignore these lines -- just needed to install and build a particular branch
import os
os.system("pip uninstall -y gradio")
os.system("pip install gradio==2.5.8a0")
### Real code is below
### -------------------------
import gradio as gr
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()
|