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