import gradio as gr def show_full_output(*args, **kwargs): model = gr.Interface.load("models/eileennoonan/resnet34_stylyst_classifier") output = model(*args, **kwargs) return output iface = gr.Interface(fn=show_full_output, inputs=gr.Interface.load("models/eileennoonan/resnet34_stylyst_classifier").get_component("input"), outputs="json") iface.launch()