RAYAuser
/

ratron-minst-2tech

Image Classification

Model card Files Files and versions

RAYAuser commited on 14 days ago

Commit

295fc63

·

verified ·

1 Parent(s): 6770fcb

Update README.md

Files changed (1) hide show

README.md +2 -2

README.md CHANGED Viewed

@@ -51,8 +51,8 @@ class SimpleFFNN(nn.Module):
 repo_id = "RAYAuser/ratron-minst-2tech"
 # Downloading the files
-ffnn_path = hf_hub_download(repo_id=repo_id, filename="ffnn_model_state.pt")
-rf_path = hf_hub_download(repo_id=repo_id, filename="random_forest_model.joblib")
 # Loading the models
 ffnn_model_loaded = SimpleFFNN()

 repo_id = "RAYAuser/ratron-minst-2tech"
 # Downloading the files
+ffnn_path = hf_hub_download(repo_id=repo_id, filename="ratron-neuronal-ffnn_model_state.pt")
+rf_path = hf_hub_download(repo_id=repo_id, filename="ratron-random_forest_model.joblib")
 # Loading the models
 ffnn_model_loaded = SimpleFFNN()