Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
|
@@ -13,6 +13,8 @@ model = load_learner(model_path)
|
|
| 13 |
def result(path):
|
| 14 |
|
| 15 |
pred,_,probability = model.predict(path)
|
|
|
|
|
|
|
| 16 |
|
| 17 |
return {pred: float(probability.max())}
|
| 18 |
|
|
|
|
| 13 |
def result(path):
|
| 14 |
|
| 15 |
pred,_,probability = model.predict(path)
|
| 16 |
+
pred = str(pred)
|
| 17 |
+
pred = pred.upper()
|
| 18 |
|
| 19 |
return {pred: float(probability.max())}
|
| 20 |
|