vuminhtue commited on
Commit
030e1d6
·
verified ·
1 Parent(s): 6f14908

Update inference.py

Browse files
Files changed (1) hide show
  1. 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(self.model_dir / "logistic_regression_model.pkl")
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