import gradio as gr
from fastai.vision.all import *

def greet(name):
    path = untar_data(URLs.PETS)
    dls = ImageDataLoaders.from_name_re(path, get_image_files(path/'images'), pat='(.+)_\d+.jpg', item_tfms=Resize(460), batch_tfms=aug_transforms(size=224, min_scale=0.75))
    learn = vision_learner(dls, models.resnet50, metrics=accuracy)
    learn.fine_tune(1)
    learn.path = Path('.')
    learn.export()
    return "Hello " + name + "!!"

iface = gr.Interface(fn=greet, inputs="text", outputs="text")
iface.launch()