vineetpasumarti's picture
Create README.md
e7fe2fe verified
|
raw
history blame
483 Bytes

You can load this model using the following code:

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