Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -33,7 +33,7 @@ def do_predict(pil_img: PIL.Image.Image):
|
|
| 33 |
# Predict probabilities
|
| 34 |
proba_df = PREDICTOR.predict_proba(df)
|
| 35 |
|
| 36 |
-
# Map columns to
|
| 37 |
pretty_dict = {}
|
| 38 |
for idx, col in enumerate(proba_df.columns):
|
| 39 |
label = CLASS_LABELS.get(idx, str(col))
|
|
|
|
| 33 |
# Predict probabilities
|
| 34 |
proba_df = PREDICTOR.predict_proba(df)
|
| 35 |
|
| 36 |
+
# Map columns to names
|
| 37 |
pretty_dict = {}
|
| 38 |
for idx, col in enumerate(proba_df.columns):
|
| 39 |
label = CLASS_LABELS.get(idx, str(col))
|