Update app.py

app.py

@@ -1,6 +1,6 @@
 # 필요한 라이브러리 임포트
 import os
-import pandas as pd
+import pandas as pd # pandas는 현재 코드에서는 직접 사용되지 않지만, 데이터 처리 관련 유틸리티로 남겨둘 수 있습니다.
 import numpy as np
 import torch
 import torch.nn as nn
@@ -62,12 +62,10 @@ class DepressionDataset(Dataset):
 # =======================================================
 print("\n--- Loading models for inference ---")
 
-# 모델 가중치가 저장된 디렉토리 (Hugging Face Spaces에 업로드할 때 생성할 폴더)
-save_dir = "saved_models"
-
-# 모델 파일 경로
-p_model_path = os.path.join(save_dir, 'p_text_best_model.bin')
-e_model_path = os.path.join(save_dir, 'e_text_best_model.bin')
+# 모델 파일 경로 (saved_models 폴더가 없으므로 루트 디렉토리에 있다고 가정)
+# 이전에 있던 save_dir 변수는 이제 필요 없습니다.
+p_model_path = 'p_text_best_model.bin' # 파일명이 루트에 바로 있다고 가정
+e_model_path = 'e_text_best_model.bin' # 파일명이 루트에 바로 있다고 가정
 
 # 모델 로딩 및 평가 모드 설정
 p_model_for_inference = None
@@ -82,7 +80,7 @@ try:
         p_model_for_inference.eval() # 평가 모드 설정
         print(f"P-model loaded successfully from {p_model_path}")
     else:
-        print(f"Warning: P-model file not found at {p_model_path}. Please ensure it's uploaded.")
+        print(f"Warning: P-model file not found at {p_model_path}. Please ensure it's uploaded to the root directory.")
 
     # 엘리 발화 모델 (E-model) 로드
     if os.path.exists(e_model_path):
@@ -92,7 +90,7 @@ try:
         e_model_for_inference.eval() # 평가 모드 설정
         print(f"E-model loaded successfully from {e_model_path}")
     else:
-        print(f"Warning: E-model file not found at {e_model_path}. Please ensure it's uploaded.")
+        print(f"Warning: E-model file not found at {e_model_path}. Please ensure it's uploaded to the root directory.")
 
 except Exception as e:
     print(f"Error loading models: {e}")
@@ -193,4 +191,3 @@ if p_model_for_inference is not None and e_model_for_inference is not None:
     ).launch() # Hugging Face Spaces에서는 share=True가 필요 없음
 else:
     print("\nGradio UI could not be launched because models failed to load. Please check model files.")
-