Spaces:
Runtime error
Runtime error
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() | |