Update model.py

model.py

@@ -1,3 +1,5 @@
+# model usage + heatmap of attention (what the model is focusing on)
+
 import cv2
 import torch
 import numpy as np
@@ -5,7 +7,7 @@ from transformers import CLIPProcessor, CLIPVisionModel
 from PIL import Image
 from torch import nn
 
-MODEL_PATH = "uno_clip_large.pth"
+MODEL_PATH = "pytorch_model.bin"
 
 class CLIPVisionClassifier(nn.Module):
     def __init__(self, num_labels):
@@ -13,7 +15,7 @@ class CLIPVisionClassifier(nn.Module):
         self.vision_model = CLIPVisionModel.from_pretrained('openai/clip-vit-large-patch14', 
                                                           attn_implementation="eager") # shows heat
         self.classifier = nn.Linear(self.vision_model.config.hidden_size, num_labels, bias=False)
-        self.dropout = nn.Dropout(0.1)
+        self.dropout = nn.Dropout(0.1) # this is not used dont worry :
         
     def forward(self, pixel_values, output_attentions=False):
         outputs = self.vision_model(pixel_values, output_attentions=output_attentions)