Update app.py
Browse files
app.py
CHANGED
|
@@ -89,7 +89,7 @@ class TextClassifier:
|
|
| 89 |
).to(self.device)
|
| 90 |
|
| 91 |
# Your existing model loading code
|
| 92 |
-
model_path = "
|
| 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)
|