Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -86,16 +86,7 @@ def predict():
|
|
| 86 |
)
|
| 87 |
# Make prediction
|
| 88 |
predictions = model.predict([img, metadata])
|
| 89 |
-
categories = [
|
| 90 |
-
"Lifestyle",
|
| 91 |
-
"Running",
|
| 92 |
-
"Other",
|
| 93 |
-
"Cleat",
|
| 94 |
-
"Sandal",
|
| 95 |
-
"Basketball",
|
| 96 |
-
"Boot",
|
| 97 |
-
"Skateboarding",
|
| 98 |
-
]
|
| 99 |
confidenceList = predictions[0].tolist()
|
| 100 |
|
| 101 |
return jsonify({"categories": categories, "confidence": confidenceList})
|
|
|
|
| 86 |
)
|
| 87 |
# Make prediction
|
| 88 |
predictions = model.predict([img, metadata])
|
| 89 |
+
categories = ['Basketball', 'Boot', 'Cleat', 'Lifestyle', 'Other', 'Running', 'Sandal', 'Skateboarding']
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 90 |
confidenceList = predictions[0].tolist()
|
| 91 |
|
| 92 |
return jsonify({"categories": categories, "confidence": confidenceList})
|