Ayamohamed
/

DiaClassModel

Image Classification

computer-vision

Model card Files Files and versions

Ayamohamed commited on Mar 28

Commit

9230778

·

verified ·

1 Parent(s): f9fbaca

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -74,7 +74,7 @@ import torch
 from huggingface_hub import hf_hub_download
 # Download model from Hugging Face Hub
-model_path = hf_hub_download(repo_id="Ayamohamed/DiaClassification", filename="dia_class_full.pth")
 # Load model
 model_hg = torch.load(model_path)

 from huggingface_hub import hf_hub_download
 # Download model from Hugging Face Hub
+model_path = hf_hub_download(repo_id="Ayamohamed/DiaClassification", filename="model.pth")
 # Load model
 model_hg = torch.load(model_path)