replace repalce return_all_tensors=True with top_k = None
Browse files- predict_dept_model.py +1 -1
predict_dept_model.py
CHANGED
|
@@ -26,7 +26,7 @@ class DepartmentPredictor:
|
|
| 26 |
model=self.model,
|
| 27 |
tokenizer=self.tokenizer,
|
| 28 |
device=self.device,
|
| 29 |
-
|
| 30 |
)
|
| 31 |
print("✅ Model and tokenizer loaded successfully.")
|
| 32 |
|
|
|
|
| 26 |
model=self.model,
|
| 27 |
tokenizer=self.tokenizer,
|
| 28 |
device=self.device,
|
| 29 |
+
top_k = None
|
| 30 |
)
|
| 31 |
print("✅ Model and tokenizer loaded successfully.")
|
| 32 |
|