Spaces:
Sleeping
Sleeping
from fastbook import * | |
import gradio as gr | |
## lets load the model | |
learn_inf = load_learner('bear_classifier_cat') | |
def bear_classifier(image): | |
pred,pred_idx,probs = learn_inf.predict(image) | |
pred_name = 'Prediction: ' , pred | |
pred_prob = 'Probability: ' , round(probs[pred_idx],4) | |
return pred_name, pred_prob | |
iface = gr.Interface(fn=bear_classifier, inputs=gr.inputs.Image(shape=(224, 224)), \ | |
outputs=["text", "number"]) | |
iface.launch(inline=False) |