Update README.md
Browse files
README.md
CHANGED
|
@@ -85,7 +85,7 @@ class ModdifiedModel(nn.Module):
|
|
| 85 |
model = ModdifiedModel(num_classes=3)
|
| 86 |
|
| 87 |
# Define the path to your model weights
|
| 88 |
-
model_weights_path = ''
|
| 89 |
|
| 90 |
state_dict = torch.hub.load_state_dict_from_url(model_weights_path, map_location='cpu')
|
| 91 |
model.load_state_dict(state_dict)
|
|
|
|
| 85 |
model = ModdifiedModel(num_classes=3)
|
| 86 |
|
| 87 |
# Define the path to your model weights
|
| 88 |
+
model_weights_path = 'https://huggingface.co/IceKhoffi/chicken-vocalization-classifier/Chiken_CNN_Disease_Detection_Model.pth'
|
| 89 |
|
| 90 |
state_dict = torch.hub.load_state_dict_from_url(model_weights_path, map_location='cpu')
|
| 91 |
model.load_state_dict(state_dict)
|