Update inference.py

inference.py

@@ -5,7 +5,7 @@ from evo_model import EvoTransformerV22
 # Load Evo model
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 evo_model = EvoTransformerV22()
-evo_model.load_state_dict(torch.load("trained_model/evo_hellaswag.pt", map_location=device))
+evo_model.load_state_dict(torch.load("trained_model_evo_hellaswag.pt", map_location=device))
 evo_model.to(device)
 evo_model.eval()