vineetpasumarti
commited on
Create README.md
Browse files
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 |
+
```
|