Alex Gonzalez
commited on
Commit
·
b34bbc3
1
Parent(s):
25e8fcd
Bugfix
Browse files
app.py
CHANGED
@@ -5,7 +5,7 @@ from PIL import Image
|
|
5 |
|
6 |
learn_inf = load_learner("export.pkl")
|
7 |
|
8 |
-
def predict(value
|
9 |
img = Image.fromarray(value).convert('LA')
|
10 |
pred,pred_idx,probs = learn_inf.predict(img)
|
11 |
return f"{pred} at {probs[pred_idx]}"
|
|
|
5 |
|
6 |
learn_inf = load_learner("export.pkl")
|
7 |
|
8 |
+
def predict(value) -> str:
|
9 |
img = Image.fromarray(value).convert('LA')
|
10 |
pred,pred_idx,probs = learn_inf.predict(img)
|
11 |
return f"{pred} at {probs[pred_idx]}"
|