import gradio as gr from datasets import load_dataset dataset = load_dataset("cifar100") dataset = load_dataset("fashion_mnist") gr.Interface.load("models/google/vit-base-patch16-224") gr.Interface.load("models/microsoft/resnet-18").launch()