Update README.md

README.md

@@ -68,8 +68,10 @@ The Modified model was evaluated on a test set.
 You can load this trained model's weights with PyTorch. For full usage examples, including audio preprocessing steps and inference, please refer to the `CHBD_Vocalization_Analysis.ipynb` notebook provided in this repository.
 
 ```python
+from huggingface_hub import hf_hub_download
 import torch
 import torch.nn as nn
+import os
 
 # Define the ModdifiedModel class
 # (You will need to copy this class definition from the CHBD_Vocalization_Analysis.ipynb file)
@@ -84,10 +86,12 @@ class ModdifiedModel(nn.Module):
 # Instantiate the model
 model = ModdifiedModel(num_classes=3)
 
-# Define the path to your model weights
-model_weights_path = 'https://huggingface.co/IceKhoffi/chicken-vocalization-classifier/Chiken_CNN_Disease_Detection_Model.pth'
+# Define each Hugging Face details
+repo_id = "IceKhoffi/chicken-vocalization-classifier"
+filename = "Chiken_CNN_Disease_Detection_Model.pth"
+model_path = hf_hub_download(repo_id=repo_id, filename=filename)
 
-state_dict = torch.hub.load_state_dict_from_url(model_weights_path, map_location='cpu')
+state_dict = torch.load(model_path, map_location='cpu')
 model.load_state_dict(state_dict)
 
 # Set model to evaluation mode