Spaces:
Sleeping
Sleeping
import gradio as gr | |
from fastai.vision.all import * | |
learn = load_learner('Landschaften.pkl') | |
labels = learn.dls.vocab | |
def predict(img): | |
pred,pred_idx,probs = learn.predict(img) | |
return {labels[i]: float(probs[i]) for i in range(len(labels))} | |
title = "Land, Berge oder Strand?" | |
description = "Ein Klassifikator für drei Landschaften in Bildern" | |
interpretation='default' | |
enable_queue=True | |
iface = gr.Interface(fn=predict, | |
inputs=gr.Image(shape=(512, 512)), | |
outputs=gr.Label(), | |
title=title, | |
description=description) | |
iface.queue() | |
iface.launch() |