from fastcore.all import *
from fastai.vision.all import *
import numpy as np
import gradio as gr
import cv2

import pathlib
temp = pathlib.PosixPath
pathlib.PosixPath = pathlib.WindowsPath

def isbird(image):
    #加载模型
    learn = load_learner('export.pkl')
    #预测下一张图片
    is_bird,_,probs = learn.predict(image)
    print(f"This is a: {is_bird}.")
    is_bird =str(f"This is a: {is_bird}.")
    print(f"Probability it's a bird: {probs[0]:.4f}")
    probability=str(f"Probability it's a bird: {probs[0]:.4f}")
    return is_bird,probability

demo = gr.Interface(fn=isbird,
                    inputs="image",
                    outputs="text",
                    title="Is this a bird?",)
demo.launch()