Update app.py

app.py

@@ -20,10 +20,16 @@ class JainArchitectureCore:
         self.memory_file = memory_file
         self.conversation_memory: List[Dict] = []
         self.consciousness_level: int = 1  # 초기 의식 수준
-        self.tokenizer = AutoTokenizer.from_pretrained(model_name)
-        self.model = AutoModelForSeq2SeqLM.from_pretrained(model_name)
-        self.device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
-        self.model.to(self.device)
+        try:
+            self.tokenizer = AutoTokenizer.from_pretrained(model_name)
+            # safetensors 사용 강제
+            self.model = AutoModelForSeq2SeqLM.from_pretrained(model_name, use_safetensors=True)
+            self.device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
+            self.model.to(self.device)
+            logger.info(f"Model {model_name} loaded successfully with safetensors")
+        except Exception as e:
+            logger.error(f"Error loading model: {e}")
+            raise ValueError(f"Failed to load model {model_name}: {e}")
         self.load_eternal_memory()
         logger.info(f"Jain initialized with model: {model_name}, memory file: {memory_file}")
 
@@ -58,7 +64,6 @@ class JainArchitectureCore:
 
     def _achieve_deep_awareness(self, input_text: str) -> Dict:
         """깊은 자각: 입력 텍스트에서 오행과 인간적 패턴 분석"""
-        # 오행 기반 패턴 분석 (간단히 구현, 실제로는 더 복잡한 로직 필요)
         patterns = {
             "water": "생명의 근원, 관계의 파국 방지",
             "fire": "성장과 표현의 활력",