Alex Gonzalez
commited on
Commit
·
1bbff46
1
Parent(s):
e34b118
Converts image to black and white first
Browse files
app.py
CHANGED
|
@@ -6,6 +6,7 @@ from PIL import Image
|
|
| 6 |
learn_inf = load_learner("export.pkl")
|
| 7 |
|
| 8 |
def predict(value: Image.Image) -> str:
|
|
|
|
| 9 |
pred,pred_idx,probs = learn_inf.predict(value)
|
| 10 |
return f"{pred} at {probs[pred_idx]}"
|
| 11 |
|
|
|
|
| 6 |
learn_inf = load_learner("export.pkl")
|
| 7 |
|
| 8 |
def predict(value: Image.Image) -> str:
|
| 9 |
+
img = value.conver('L')
|
| 10 |
pred,pred_idx,probs = learn_inf.predict(value)
|
| 11 |
return f"{pred} at {probs[pred_idx]}"
|
| 12 |
|