Updated README.md

README.md

@@ -60,7 +60,9 @@ config = {
     "use_layer_norm": True
 }
 model = BERT_FFNN(**config)
-model.load_state_dict(torch.load("pytorch_model.bin", map_location="cpu"))
+model_path = hf_hub_download(repo_id="NeuralNest05/emo-detector", filename="pytorch_model.bin")
+model.load_state_dict(torch.load(model_path, map_location=DEVICE))
+model.to(DEVICE)
 model.eval()
 
 # Example prediction