Update model_loader.py

model_loader.py

@@ -2,7 +2,7 @@ import os
 import torch
 import torchvision
 
-def build_SurgFM(nclasses: int = 2, pretrained: bool = True, pretrained_weights = None):
+def build_LemonFM(nclasses: int = 2, pretrained: bool = True, pretrained_weights = None):
 
 
     #net of ConvNext