Spaces:
Sleeping
Sleeping
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 |
|