FIX: pretrained_models and offline

app.py

@@ -14,9 +14,9 @@ def load_model(detection_type):
 
     device = torch.device("cpu")
 
-    processor = AutoProcessor.from_pretrained("openai/clip-vit-large-patch14")
-    clip_model = CLIPVisionModel.from_pretrained("openai/clip-vit-large-patch14", output_attentions=True)
-
+    processor = AutoProcessor.from_pretrained("clip-vit-large-patch14")
+    clip_model = CLIPVisionModel.from_pretrained("clip-vit-large-patch14", output_attentions=True)
+    
     model_path = f"pretrained_models/{detection_type}/clip_weights.pth"
     checkpoint = torch.load(model_path, map_location="cpu")
     input_dim = checkpoint["linear.weight"].shape[1]
@@ -32,7 +32,7 @@ def process_image(image, detection_type):
     
     results = detect_image(image, processor, clip_model, detection_model)
 
-    pred_score = results["pred_score"]
+    pred_score = 1 - results["pred_score"]
     attn_map = results["attn_map"]
 
     return pred_score, attn_map

clip-vit-large-patch14

@@ -0,0 +1 @@
+Subproject commit 32bd64288804d66eefd0ccbe215aa642df71cc41