Spaces:
Sleeping
Sleeping
SalmanHabeeb
commited on
Commit
·
16f509e
1
Parent(s):
0ede29c
Corrected variable name
Browse files
app.py
CHANGED
@@ -10,7 +10,7 @@ model = load_learner("model.pkl")
|
|
10 |
|
11 |
def predict(img):
|
12 |
pred, idx, probs = model.predict(img)
|
13 |
-
dict1 = dict(zip(categories, map(float,
|
14 |
dict1 = dict(sorted(dict1.items(), key=lambda item : item[1]))
|
15 |
output = {key:dict1[key] for key in dict1.keys()[-3:]}
|
16 |
sum = 0
|
|
|
10 |
|
11 |
def predict(img):
|
12 |
pred, idx, probs = model.predict(img)
|
13 |
+
dict1 = dict(zip(categories, map(float, probs)))
|
14 |
dict1 = dict(sorted(dict1.items(), key=lambda item : item[1]))
|
15 |
output = {key:dict1[key] for key in dict1.keys()[-3:]}
|
16 |
sum = 0
|