Update app.py

app.py

@@ -1,22 +1,11 @@
 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