ApsidalSolid4 commited on
Commit
1351283
·
verified ·
1 Parent(s): cc3e103

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -89,7 +89,7 @@ class TextClassifier:
89
  ).to(self.device)
90
 
91
  # Your existing model loading code
92
- model_path = "model.pt"
93
  if os.path.exists(model_path):
94
  logger.info(f"Loading custom model from {model_path}")
95
  checkpoint = torch.load(model_path, map_location=self.device)
 
89
  ).to(self.device)
90
 
91
  # Your existing model loading code
92
+ model_path = "model_20250209_184929_acc1.0000.pt"
93
  if os.path.exists(model_path):
94
  logger.info(f"Loading custom model from {model_path}")
95
  checkpoint = torch.load(model_path, map_location=self.device)