bear_classifier / app.py
李根赢
edit app.py
dc7601d
raw
history blame
644 Bytes
# AUTOGENERATED! DO NOT EDIT! File to edit: gdrive/MyDrive/Colab Notebooks/Untitled8.ipynb.
# %% auto 0
__all__ = ['path', 'learn_inf', 'btn_upload', 'pred', 'pred_idx', 'probs']
# %% gdrive/MyDrive/Colab Notebooks/Untitled8.ipynb 24
from fastai.vision.widgets import *
from fastai.vision.all import *
import gradio as gr
# %% gdrive/MyDrive/Colab Notebooks/Untitled8.ipynb 25
# path = Path('bears')
learn_inf = load_learner('export.pkl')
# %% gdrive/MyDrive/Colab Notebooks/Untitled8.ipynb 26
btn_upload = widgets.FileUpload()
btn_upload
# %% gdrive/MyDrive/Colab Notebooks/Untitled8.ipynb 30
pred,pred_idx,probs = learn_inf.predict(img)