Update app.py

app.py

@@ -1,10 +1,22 @@
 import json, os
 import numpy as np
 from PIL import Image
+import torch
+
+def peek_out_dim(pt_path):
+    sd = clean_state_dict(torch.load(pt_path, map_location="cpu"))
+    # timm classifier keys are usually these:
+    for k in ["classifier.weight", "head.weight", "fc.weight"]:
+        if k in sd:
+            return sd[k].shape[0]
+    return None
+
+print("✅ image checkpoint out dim:", peek_out_dim("best_scin_image.pt"))
+print("✅ text checkpoint out dim :", peek_out_dim("best_scin_text.pt"))
+
 
 
 
-import torch
 import torch.nn as nn
 import timm
 from timm.data import resolve_model_data_config, create_transform