Spaces:
Runtime error
Runtime error
File size: 956 Bytes
5db7e60 b7b3dd2 5db7e60 a4a3b4f 5db7e60 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 |
import gradio as gr
import os
pkl = "all_20210912_res34.pkl"
from fastai.vision.all import *
from fastai.vision.widgets import *
import jaconv
import pathlib
plt = platform.system()
if plt == 'Linux': pathlib.WindowsPath = pathlib.PosixPath
model_inf = load_learner(pkl)
print(os.getcwd())
def kinoko_uranai(img):
replace_dic = {"_ッロウッ":" (group)","ー":""}
result_dic = {}
pred_class, pred_idxs, outputs = model_inf.predict(img)
top_5_conf, i = outputs.topk(5)
itr = 0
classes = model_inf.dls.vocab
result_dic = {}
for x in i:
kwamei = jaconv.alphabet2kata(classes[x.item()].lower())
for k,v in replace_dic.items():
kwamei = kwamei.replace(k,v)
result_dic[kwamei] = str(round(top_5_conf[itr].item(),2))
itr=itr+1
return result_dic
outputs = gr.outputs.Label(num_top_classes=5)
iface = gr.Interface(fn=kinoko_uranai, inputs="image", outputs=outputs).launch(debug=True) |