import gradio as gr import torch pipe = torch.hub.load('nateraw/image-generation:main', 'styleganv3', force_reload=True) def fn(text): return text iface = gr.Interface(fn, inputs='text', outputs='text') iface.launch()