mr-kush commited on
Commit
e787a7e
·
verified ·
1 Parent(s): 7583127

replace repalce return_all_tensors=True with top_k = None

Browse files
Files changed (1) hide show
  1. 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
- return_all_scores=True
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