vineetpasumarti's picture
Create README.md
e7fe2fe verified
|
raw
history blame
483 Bytes
## You can load this model using the following code:
```python
from huggingface_hub import hf_hub_download
import torch
# Specify the repository and the filename of the model you want to load
repo_id = "{organization_name}/{repo_name}" # Replace with your repo name
filename = "{path_name}"
model_path = hf_hub_download(repo_id=repo_id, filename=filename)
# Load the model using torch
model_test = torch.load(model_path)
model_test.eval() # Set the model to evaluation mode
```