PerceptCLIP
/

PerceptCLIP_Emotions

emotion_prediction

computer_vision

perceptual_tasks

Model card Files Files and versions

Amitz244 commited on Mar 19, 2025

Commit

1f44e9b

·

verified ·

1 Parent(s): 149ee8b

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -36,7 +36,7 @@ from PIL import Image
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 # Load model
-model = torch.load("EmoSet_clip_Lora_16.0R_8.0alphaLora_32_batch_0.0001_headmlp.pth").to(device).eval()
 # Emotion label mapping
 idx2label = {

 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 # Load model
+model = torch.load("perceptCLIP_Emotions.pth").to(device).eval()
 # Emotion label mapping
 idx2label = {