File size: 525 Bytes
0920a68 4ffe84c 0920a68 4ffe84c 0920a68 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 | 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()
|