kdevoe commited on
Commit
61cfeda
1 Parent(s): 78497a7

Update inference.py

Browse files
Files changed (1) hide show
  1. inference.py +1 -2
inference.py CHANGED
@@ -26,8 +26,7 @@ def inference():
26
  saved_model_path = './'
27
  model = torch.load(saved_model_path + 'fine-tuned-model.pt', map_location=torch.device(device))
28
  tokenizer = torch.load(saved_model_path + 'fine-tuned-tokenizer.pt', map_location=torch.device(device))
29
-
30
- model.eval()
31
 
32
  if __name__ == "__main__":
33
  inference()
 
26
  saved_model_path = './'
27
  model = torch.load(saved_model_path + 'fine-tuned-model.pt', map_location=torch.device(device))
28
  tokenizer = torch.load(saved_model_path + 'fine-tuned-tokenizer.pt', map_location=torch.device(device))
29
+ model.eval()
 
30
 
31
  if __name__ == "__main__":
32
  inference()