Update app.py

app.py

@@ -3,8 +3,12 @@ import torch
 from transformers import AutoTokenizer, AutoModelForSequenceClassification, pipeline
 
 # 모델 및 토크나이저 로드
-model = AutoModelForSequenceClassification.from_pretrained("pytorch_model_ethics8multilable_acc8997.bin")
-tokenizer = AutoTokenizer.from_pretrained("tokenizer.json")
+
+model_directory = "."  # 현재 작업 디렉토리를 사용하여 모델 로드
+model = AutoModelForSequenceClassification.from_pretrained(model_directory)
+tokenizer = AutoTokenizer.from_pretrained(model_directory)
+# model = AutoModelForSequenceClassification.from_pretrained("pytorch_model_ethics8multilable_acc8997.bin")
+# tokenizer = AutoTokenizer.from_pretrained("tokenizer.json")
 
 # 추론 파이프라인 설정
 inference_pipeline = pipeline(