waseoke commited on
Commit
067bb59
·
verified ·
1 Parent(s): fb24e6e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -23,15 +23,15 @@ def execute_script(script_name):
23
 
24
  def main():
25
  # # Step 0: 모델 학습
26
- print("Step 1: 모델 학습 중...")
27
  execute_script("train_model.py")
28
 
29
  # Step 1: 쇼핑물 상품과 사용자 임베딩
30
- print("Step 2: 쇼핑물 상품과 사용자 임베딩...")
31
  execute_script("embed_data.py")
32
 
33
  # Step 2: product_model.pth 불러오기
34
- print("Step 3: product_model.pth 불러오는 중...")
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 입력