Update README.md

README.md

@@ -54,7 +54,7 @@ from model import LVL  # Replace with your module or package
 
 # Load model
 model = LVL()
-model.load_state_dict(torch.load("pytorch_model.bin", map_location="cpu"))
+model.load_state_dict(torch.load("model_fn.bin"))
 model.eval()
 
 # Text preprocessing