Spaces:
Sleeping
Sleeping
# AUTOGENERATED! DO NOT EDIT! File to edit: ../app.ipynb. | |
# %% auto 0 | |
__all__ = ['temp', 'learn', 'categories', 'image', 'label', 'examples', 'iface', 'ECG', 'classify_image'] | |
# %% ../app.ipynb 2 | |
from fastai.vision.all import* | |
import gradio as gr | |
def ECG(x): | |
return x[0].isupper() | |
# %% ../app.ipynb 4 | |
# import pathlib | |
# temp = pathlib.PosixPath | |
# pathlib.PosixPath = pathlib.WindowsPath | |
# from pathlib import Path | |
learn = load_learner('model1.pkl') | |
# %% ../app.ipynb 6 | |
categories = ('bradycardia','tachycardia') | |
def classify_image(img): | |
pred,idx,probs = learn.predict(img) | |
return dict(zip(categories, map(float,probs))) | |
# %% ../app.ipynb 8 | |
image = gr.inputs.Image(shape=(192,192)) | |
label = gr.outputs.Label() | |
examples = ['bradycardia-ECG.jpg'] | |
iface = gr.Interface(fn = classify_image, inputs = image, outputs = label, examples = examples) | |
iface.launch() | |