diff --git a/AI/app/snack_db.py b/AI/app/snack_db.py index 36b67ae..b66b7ac 100644 --- a/AI/app/snack_db.py +++ b/AI/app/snack_db.py @@ -238,7 +238,7 @@ def sortkey_est(pred): for pred in top_predictions: snack_id = int(pred.iid) - snack_title = snack_db[snack_db["snack_id"] == snack_id]["이름"].tolist() + snack_title = snack_db[snack_db["snack_id"] == snack_id]["name"].tolist() snack_rating = pred.est top_snack_preds.append(round(snack_rating, 2))