vineetpasumarti commited on
Commit
e7fe2fe
·
verified ·
1 Parent(s): 0d48314

Create README.md

Browse files
Files changed (1) hide show
  1. README.md +16 -0
README.md CHANGED
@@ -0,0 +1,16 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ## You can load this model using the following code:
2
+
3
+ ```python
4
+ from huggingface_hub import hf_hub_download
5
+ import torch
6
+
7
+ # Specify the repository and the filename of the model you want to load
8
+ repo_id = "{organization_name}/{repo_name}" # Replace with your repo name
9
+ filename = "{path_name}"
10
+
11
+ model_path = hf_hub_download(repo_id=repo_id, filename=filename)
12
+
13
+ # Load the model using torch
14
+ model_test = torch.load(model_path)
15
+ model_test.eval() # Set the model to evaluation mode
16
+ ```