diff --git a/BERT/test_metric.py b/BERT/test_metric.py
index ba2cee7..f1fb4dd 100644
--- a/BERT/test_metric.py
+++ b/BERT/test_metric.py
@@ -42,7 +42,7 @@ if __name__ == '__main__':
     data_loader = DataLoader(dataset, batch_size=batch_size, shuffle=False, num_workers=1)
 
     # 加载模型和检查点
-    model = BertForSequenceClassification.from_pretrained('./BERT_training_output/checkpoint-35544')
+    model = BertForSequenceClassification.from_pretrained('./BERT_training_output/checkpoint-35544'， num_labels = 6)
 
     # 运行推断
     model.eval()