TAGCH
Add application file
126bbcd
from fastai.vision.all import *
import gradio as gr
# Import untar_data
path = untar_data(URLs.PETS)/'images'
def is_BullTerrier(x):
"""Check if it's a Bull Terrier dog."""
return x[0].isupper()
# Define item transforms
item_tfms = Resize(192)
# Create ImageDataLoaders
dls = ImageDataLoaders.from_name_func('.',
get_image_files(path), valid_pct=0.2,
seed=42, label_func=is_BullTerrier,
item_tfms=item_tfms)
# Create vision learner
learn = vision_learner(dls, resnet18, metrics=error_rate)
learn.fine_tune(3)
dls.show_batch()
# Use is_BullTerrier instead of greet in Gradio interface
iface = gr.Interface(fn=is_BullTerrier, inputs="text", outputs="text")
iface.launch()