import gradio as gr from distilabel import DistilAbel distilabel = DistilAbel(model_name="distilbert-base-uncased") def label_text(text): labels = distilabel.label(text) return labels iface = gr.Interface( fn=label_text, inputs="text", outputs="text", title="Distilabel Demo", description="Enter text to label:" ) if __name__ == "__main__": iface.launch()