Update README.md

README.md

@@ -39,10 +39,17 @@ from torchvision import transforms
 import torch
 from PIL import Image
 from huggingface_hub import hf_hub_download
+import importlib.util
 
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 
-# Load model
+# Load model class
+class_path = hf_hub_download(repo_id="PerceptCLIP/PerceptCLIP_Emotions", filename="modeling.py")
+spec = importlib.util.spec_from_file_location("clip_lora_model", class_path)
+clip_lora_model = importlib.util.module_from_spec(spec)
+spec.loader.exec_module(clip_lora_model)
+
+# Load pretrained model
 model_path = hf_hub_download(repo_id="PerceptCLIP/PerceptCLIP_Emotions", filename="perceptCLIP_Emotions.pth")
 model = torch.load(model_path).to(device).eval()