Update model.py

model.py

@@ -7,4 +7,4 @@ def load_model():
     model = EfficientNet.from_name("efficientnet-b0", num_classes=len(class_names))
     model.load_state_dict(torch.load("efnet_fracture_classifier.pth", map_location="cpu"))
     model.eval()
-    return model
+    return model