unit_test.py

import gradio as gr
import os
HF_TOKEN = os.environ.get('HF_TOKEN')
hf_writer = gr.HuggingFaceDatasetSaver(HF_TOKEN, "image-classification-mistakes")
def image_classifier(inp):
    return {'cat': 0.3, 'dog': 0.7}
demo = gr.Interface(fn=image_classifier, inputs="image", outputs="label",
                allow_flagging="manual", flagging_callback=hf_writer)

# Launch the interface
demo.launch()