Update inference.py
Browse files- 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()
|