Spaces:
Sleeping
Sleeping
๊ฐ๋ฏผ๊ท commited on
Commit ยท
c95db27
1
Parent(s): e18fae2
Fix: Absolute path for data loading and add import os
Browse files- backend/src/logic.py +10 -5
backend/src/logic.py
CHANGED
|
@@ -8,6 +8,7 @@ import pickle
|
|
| 8 |
from datetime import datetime, timedelta, timezone
|
| 9 |
from supabase import create_client
|
| 10 |
import re
|
|
|
|
| 11 |
from collections import Counter
|
| 12 |
|
| 13 |
# ==========================================
|
|
@@ -243,18 +244,22 @@ def load_resources():
|
|
| 243 |
global method_map, recipes_by_ingredient, ing_method_counts, ing_cat_counts, total_method_counts, total_cat_counts, TOTAL_RECIPES
|
| 244 |
|
| 245 |
print("Loading resources... (This may take a while)")
|
|
|
|
|
|
|
|
|
|
|
|
|
| 246 |
# mmap='r' ์ต์
์ผ๋ก ๋ฉ๋ชจ๋ฆฌ ์ฌ์ฉ๋ ์ต์ํ (๋์คํฌ์์ ์ง์ ์ฝ์)
|
| 247 |
-
w2v_model = Word2Vec.load("models/w2v.model", mmap='r')
|
| 248 |
-
d2v_model = Doc2Vec.load("models/d2v.model", mmap='r')
|
| 249 |
-
df_temp = pd.read_csv("data/final_recipe_data.csv")
|
| 250 |
df_temp['์ฌ๋ฃํ ํฐ'] = df_temp['์ฌ๋ฃํ ํฐ'].apply(literal_eval)
|
| 251 |
df = df_temp # Assign to global
|
| 252 |
|
| 253 |
-
with open("data/stats.pkl", "rb") as f:
|
| 254 |
stats = pickle.load(f)
|
| 255 |
|
| 256 |
try:
|
| 257 |
-
price_df = pd.read_csv("data/price_rank.csv", encoding='utf-8-sig')
|
| 258 |
price_df.columns = price_df.columns.str.strip()
|
| 259 |
price_map = dict(zip(price_df['ingredient'], price_df['rank']))
|
| 260 |
except:
|
|
|
|
| 8 |
from datetime import datetime, timedelta, timezone
|
| 9 |
from supabase import create_client
|
| 10 |
import re
|
| 11 |
+
import os
|
| 12 |
from collections import Counter
|
| 13 |
|
| 14 |
# ==========================================
|
|
|
|
| 244 |
global method_map, recipes_by_ingredient, ing_method_counts, ing_cat_counts, total_method_counts, total_cat_counts, TOTAL_RECIPES
|
| 245 |
|
| 246 |
print("Loading resources... (This may take a while)")
|
| 247 |
+
|
| 248 |
+
# ๊ธฐ์ค ๊ฒฝ๋ก ์ค์ (ํ์ฌ ํ์ผ ์์น logic.py ๊ธฐ์ค ์์ ํด๋)
|
| 249 |
+
base_dir = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
|
| 250 |
+
|
| 251 |
# mmap='r' ์ต์
์ผ๋ก ๋ฉ๋ชจ๋ฆฌ ์ฌ์ฉ๋ ์ต์ํ (๋์คํฌ์์ ์ง์ ์ฝ์)
|
| 252 |
+
w2v_model = Word2Vec.load(os.path.join(base_dir, "models/w2v.model"), mmap='r')
|
| 253 |
+
d2v_model = Doc2Vec.load(os.path.join(base_dir, "models/d2v.model"), mmap='r')
|
| 254 |
+
df_temp = pd.read_csv(os.path.join(base_dir, "data/final_recipe_data.csv"))
|
| 255 |
df_temp['์ฌ๋ฃํ ํฐ'] = df_temp['์ฌ๋ฃํ ํฐ'].apply(literal_eval)
|
| 256 |
df = df_temp # Assign to global
|
| 257 |
|
| 258 |
+
with open(os.path.join(base_dir, "data/stats.pkl"), "rb") as f:
|
| 259 |
stats = pickle.load(f)
|
| 260 |
|
| 261 |
try:
|
| 262 |
+
price_df = pd.read_csv(os.path.join(base_dir, "data/price_rank.csv"), encoding='utf-8-sig')
|
| 263 |
price_df.columns = price_df.columns.str.strip()
|
| 264 |
price_map = dict(zip(price_df['ingredient'], price_df['rank']))
|
| 265 |
except:
|