import gradio as gr

demo = gr.load("MuntasirHossain/distilbert-finetuned-ag-news", src="models")

demo.launch()