Update macbert/infer_all.py

macbert/infer_all.py

@@ -88,10 +88,10 @@ def test_epoch(travel_model, name_model, epoch, dataloader, tokenizer):
     return travel_predictions, travel_probs, name_predictions, name_probs, sms_ids
 
 def inference():
-    travel_checkpoint_file = '/home/jchsiao/Desktop/macbert/checkpoints/saved_checkpoints/travel_checkpoint15_train8000.pth.tar'
-    name_checkpoint_file = '/home/jchsiao/Desktop/macbert/checkpoints/saved_checkpoints/name_checkpoint17_train9000.pth.tar'
-    ann_file_test = r'/home/jchsiao/Desktop/macbert/dataset/datagame_sms_stage1(in).csv'
-    output_file = r'/home/jchsiao/Desktop/macbert/both_macbertBase_20250731_2.csv'
+    travel_checkpoint_file = 'checkpoints/saved_checkpoints/travel_checkpoint15_train8000.pth.tar'
+    name_checkpoint_file = 'checkpoints/saved_checkpoints/name_checkpoint17_train9000.pth.tar'
+    ann_file_test = r'dataset/datagame_sms_stage1(in).csv'
+    output_file = r'/both_macbertBase_20250731_2.csv'
     cache_dir = 'cache'
 
     model_cfg = {