Views
No views yet
1import torch
2from huggingface_hub import hf_hub_download
3
4# Download checkpoint
5checkpoint_path = hf_hub_download(
6 repo_id="RaghavIITM/enhanced-bilstm-nppe2",
7 filename="best_model.ckpt"
8)
9
10# Load model
11checkpoint = torch.load(checkpoint_path, map_location='cpu')
12print("Model loaded successfully!")