Fix: 컬럼명 변경

lmHOLO · Apr 5, 2023 · 2c223f3 · 2c223f3
1 parent 81b6cb7
commit 2c223f3
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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))