Spaces:
Sleeping
Sleeping
import json | |
import gradio as gr | |
import yolov5 | |
model = yolov5.load('model/Cinderella.pt') | |
#てすと | |
def inference(gr_input): | |
with open("json/idol_classes.json", "r")as f: | |
dic = json.load(f) | |
predictions = model(gr_input).pred[0] | |
name = [x for x in dic.values()] | |
idols = [] | |
for i in predictions[:, 5]: # 検出 | |
idols.append(name[int(i.item())]) | |
if len(idols) == 0: | |
return "該当するクラスが存在しません" | |
else: | |
text = "\n".join(idols) | |
return text | |
with gr.Blocks() as app: | |
inputs = gr.inputs.Image() | |
output = gr.Textbox() | |
gr.Markdown( | |
"# <center> デレマス画像分類器\n" | |
"## <center> シンデレラガールズのキャラの画像を貼ると、機械学習で判別できるモデルです\n" | |
) | |
inputs.change(inference, inputs, out) | |
demo.launch() | |