import gradio as gr

gr.Interface.load("models/lvwerra/distilbert-imdb").launch()