Update README.md

README.md

@@ -31,11 +31,13 @@ To use the model for inference:
 from torchvision import transforms
 import torch
 from PIL import Image
+from huggingface_hub import hf_hub_download
 
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 
 # Load model
-model = torch.load("perceptCLIP_Emotions.pth").to(device).eval()
+model_path = hf_hub_download(repo_id="PerceptCLIP/PerceptCLIP_Emotions", filename="perceptCLIP_Emotions.pth")
+model = torch.load(model_path).to(device).eval()
 
 # Emotion label mapping
 idx2label = {