import gradio as gr

from huggan.pytorch.pix2pix.modeling_pix2pix import GeneratorUNet

model = GeneratorUNet.from_pretrained('huggan/pix2pix-edge2shoes')

def dummy_fn(text):
    return "Looks like the model is loaded!"

gr.Interface(dummy_fn, inputs='text', outputs='text').launch()