Update app.py

app.py

@@ -8,7 +8,7 @@ model_name = "SAVSNET/PetBERT_ICD"
 tokenizer = AutoTokenizer.from_pretrained(model_name)
 model = AutoModelForSequenceClassification.from_pretrained(model_name)
 
-# 2. 라벨 목록 (ICD 코드 등은 예시, 필요시 HuggingFace card 참조)
+# 2. 라벨 목록 (ICD 코드 등은 예시)
 LABELS = {
     0: "소화기 질환",
     1: "호흡기 질환",
@@ -18,6 +18,9 @@ LABELS = {
     5: "기타 질병"
 }
 
+# 라벨 출력 (Logs에서 확인 가능)
+print("🔍 라벨 매핑:", LABELS)
+
 # 3. 예측 함수 정의
 def predict(text):
     try: