Update app.py

app.py

@@ -70,8 +70,8 @@ class AccidentAnalysisModel:
             # 2. 분류 모델 로드
             classification_dir = f"{self.model_repo}/classification_model"  # 하위 폴더 지정
             print(f"분류 모델 로드 중: {classification_dir}")
-            self.classification_tokenizer = AutoTokenizer.from_pretrained(classification_dir)
-            self.classification_model = AutoModelForSequenceClassification.from_pretrained(classification_dir)
+            self.classification_tokenizer = AutoTokenizer.from_pretrained("jennalee1385/accident_classification")
+            self.classification_model = AutoModelForSequenceClassification.from_pretrained("jennalee1385/accident_classification")
             self.classification_model.to(self.device)
         
             # 3. 레이블 맵 로드