Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -23,15 +23,15 @@ def execute_script(script_name):
|
|
23 |
|
24 |
def main():
|
25 |
# # Step 0: 모델 학습
|
26 |
-
print("
|
27 |
execute_script("train_model.py")
|
28 |
|
29 |
# Step 1: 쇼핑물 상품과 사용자 임베딩
|
30 |
-
print("
|
31 |
execute_script("embed_data.py")
|
32 |
|
33 |
# Step 2: product_model.pth 불러오기
|
34 |
-
print("
|
35 |
model = load_trained_model("product_model.pth")
|
36 |
|
37 |
# Step 3: 추천을 위한 사용자 ID 입력
|
|
|
23 |
|
24 |
def main():
|
25 |
# # Step 0: 모델 학습
|
26 |
+
print("모델 학습 중...")
|
27 |
execute_script("train_model.py")
|
28 |
|
29 |
# Step 1: 쇼핑물 상품과 사용자 임베딩
|
30 |
+
print("쇼핑물 상품과 사용자 임베딩...")
|
31 |
execute_script("embed_data.py")
|
32 |
|
33 |
# Step 2: product_model.pth 불러오기
|
34 |
+
print("product_model.pth 불러오는 중...")
|
35 |
model = load_trained_model("product_model.pth")
|
36 |
|
37 |
# Step 3: 추천을 위한 사용자 ID 입력
|