Spaces:
Sleeping
Sleeping
Sungmin Son
commited on
Commit
·
542596e
1
Parent(s):
fd7ac62
No comment
Browse files- Untitled-1.ipynb +1 -1
- app +0 -30
Untitled-1.ipynb
CHANGED
|
@@ -213,7 +213,7 @@
|
|
| 213 |
},
|
| 214 |
{
|
| 215 |
"cell_type": "code",
|
| 216 |
-
"execution_count":
|
| 217 |
"metadata": {},
|
| 218 |
"outputs": [],
|
| 219 |
"source": [
|
|
|
|
| 213 |
},
|
| 214 |
{
|
| 215 |
"cell_type": "code",
|
| 216 |
+
"execution_count": 12,
|
| 217 |
"metadata": {},
|
| 218 |
"outputs": [],
|
| 219 |
"source": [
|
app
DELETED
|
@@ -1,30 +0,0 @@
|
|
| 1 |
-
# AUTOGENERATED! DO NOT EDIT! File to edit: ../../Untitled-1.ipynb.
|
| 2 |
-
|
| 3 |
-
# %% auto 0
|
| 4 |
-
__all__ = ['model_path', 'learn', 'categories', 'is_cat', 'classify_image']
|
| 5 |
-
|
| 6 |
-
# %% ../../Untitled-1.ipynb 0
|
| 7 |
-
import gradio as gr
|
| 8 |
-
from fastai.vision.all import *
|
| 9 |
-
def is_cat(x): return x[0].isupper()
|
| 10 |
-
|
| 11 |
-
# %% ../../Untitled-1.ipynb 2
|
| 12 |
-
model_path = Path('model.pkl')
|
| 13 |
-
learn = load_learner(model_path)
|
| 14 |
-
|
| 15 |
-
# %% ../../Untitled-1.ipynb 4
|
| 16 |
-
categories = ('dog', 'cat')
|
| 17 |
-
|
| 18 |
-
def classify_image(img):
|
| 19 |
-
pred,idx,probs = learn.predict(img)
|
| 20 |
-
return dict(zip(categories, map(float,probs)))
|
| 21 |
-
|
| 22 |
-
|
| 23 |
-
# %% ../../Untitled-1.ipynb 6
|
| 24 |
-
gr.Interface(fn = classify_image,
|
| 25 |
-
inputs="image",
|
| 26 |
-
outputs="label",
|
| 27 |
-
title = "Dog or Cat",
|
| 28 |
-
description="Sample workflow",
|
| 29 |
-
examples = ['dog.jpg']
|
| 30 |
-
).launch()
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|