ApsidalSolid4 commited on
Commit
154774e
·
verified ·
1 Parent(s): 98f1c59

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -99,7 +99,7 @@ class TextClassifier:
99
  num_labels=2
100
  ).to(self.device)
101
 
102
- model_path = "model.pt"
103
  if os.path.exists(model_path):
104
  logger.info(f"Loading custom model from {model_path}")
105
  checkpoint = torch.load(model_path, map_location=self.device)
 
99
  num_labels=2
100
  ).to(self.device)
101
 
102
+ model_path = "model_20250209_184929_acc1.0000.pt"
103
  if os.path.exists(model_path):
104
  logger.info(f"Loading custom model from {model_path}")
105
  checkpoint = torch.load(model_path, map_location=self.device)