Update app.py

app.py

@@ -56,8 +56,8 @@ dinov2 = torch.hub.load('facebookresearch/dinov2', 'dinov2_vitg14_reg')
 print("Loading custom pathology checkpoint...")
 #ours = torch.load("/data/linears/teacher_checkpoint.pth")
 
-checkpoint = torch.load("./teacher_checkpoint_load.pt")
-
+#checkpoint = torch.load("./teacher_checkpoint_load.pt")
+checkpoint = torch.hub.load_state_dict_from_url("https://huggingface.co/SophontAI/OpenMidnight/resolve/main/teacher_checkpoint_load.pt")
 
 
 new_shape = checkpoint["pos_embed"]