zachwormgoor@gmail.com
commited on
Commit
•
0917cd6
1
Parent(s):
a2c132a
fix
Browse files
app.py
CHANGED
|
@@ -3,7 +3,7 @@
|
|
| 3 |
|
| 4 |
|
| 5 |
|
| 6 |
-
__all__ = ['install', 'learn', 'categories', '
|
| 7 |
|
| 8 |
|
| 9 |
|
|
@@ -47,7 +47,7 @@ learn = load_learner('model.pkl')
|
|
| 47 |
|
| 48 |
categories = ('stock', 'amateur')
|
| 49 |
|
| 50 |
-
def
|
| 51 |
pred,idx,probs = learn.predict(img)
|
| 52 |
return dict(zip(categories, map(float,probs)))
|
| 53 |
|
|
|
|
| 3 |
|
| 4 |
|
| 5 |
|
| 6 |
+
__all__ = ['install', 'learn', 'categories', 'classify_image', 'image', 'label', 'examples', 'intf']
|
| 7 |
|
| 8 |
|
| 9 |
|
|
|
|
| 47 |
|
| 48 |
categories = ('stock', 'amateur')
|
| 49 |
|
| 50 |
+
def classify_image(img):
|
| 51 |
pred,idx,probs = learn.predict(img)
|
| 52 |
return dict(zip(categories, map(float,probs)))
|
| 53 |
|