Update inference.py
Browse files- inference.py +1 -1
inference.py
CHANGED
|
@@ -67,7 +67,7 @@ class SentimentClassifier:
|
|
| 67 |
self.hook = self.qwen_model.final_norm.register_forward_hook(self._hook_fn)
|
| 68 |
|
| 69 |
# Load LR model
|
| 70 |
-
self.lr_model = joblib.load(
|
| 71 |
|
| 72 |
print(f"✅ Loaded successfully! Test Accuracy: {self.config['test_accuracy']*100:.2f}%")
|
| 73 |
|
|
|
|
| 67 |
self.hook = self.qwen_model.final_norm.register_forward_hook(self._hook_fn)
|
| 68 |
|
| 69 |
# Load LR model
|
| 70 |
+
self.lr_model = joblib.load("logistic_regression_model.pkl")
|
| 71 |
|
| 72 |
print(f"✅ Loaded successfully! Test Accuracy: {self.config['test_accuracy']*100:.2f}%")
|
| 73 |
|