import os
import gradio as gr

HF_TOKEN = os.getenv('HF_TOKEN')

hf_writer = gr.HuggingFaceDatasetSaver(HF_TOKEN, "crowdsourced-speech-demo")

iface = gr.Interface.load(
    "models/facebook/wav2vec2-base-960h",
    inputs="mic",
    flagging_callback=hf_writer)

iface.launch()