Update README.md
Browse files
README.md
CHANGED
|
@@ -51,8 +51,8 @@ class SimpleFFNN(nn.Module):
|
|
| 51 |
repo_id = "RAYAuser/ratron-minst-2tech"
|
| 52 |
|
| 53 |
# Downloading the files
|
| 54 |
-
ffnn_path = hf_hub_download(repo_id=repo_id, filename="ffnn_model_state.pt")
|
| 55 |
-
rf_path = hf_hub_download(repo_id=repo_id, filename="random_forest_model.joblib")
|
| 56 |
|
| 57 |
# Loading the models
|
| 58 |
ffnn_model_loaded = SimpleFFNN()
|
|
|
|
| 51 |
repo_id = "RAYAuser/ratron-minst-2tech"
|
| 52 |
|
| 53 |
# Downloading the files
|
| 54 |
+
ffnn_path = hf_hub_download(repo_id=repo_id, filename="ratron-neuronal-ffnn_model_state.pt")
|
| 55 |
+
rf_path = hf_hub_download(repo_id=repo_id, filename="ratron-random_forest_model.joblib")
|
| 56 |
|
| 57 |
# Loading the models
|
| 58 |
ffnn_model_loaded = SimpleFFNN()
|