Update app.py

app.py

@@ -132,7 +132,7 @@ if len(file_upload) == 1:
             model = EfficientNetBN(
                 "efficientnet-b0", spatial_dims=3, in_channels=1, num_classes=3)
             model.load_state_dict(torch.load(
-                'MCEBNfold5.pth', map_location='cpu'))
+                'MCEBNfold2.pth', map_location='cpu'))
             model.eval()
             prediction = model(test_images.unsqueeze(1))
             pred = prediction.argmax(dim=1).item()
@@ -310,7 +310,7 @@ if len(file_upload) > 1:
             model = EfficientNetBN(
                 "efficientnet-b0", spatial_dims=3, in_channels=1, num_classes=3)
             model.load_state_dict(torch.load(
-                'MCEBNfold5.pth', map_location='cpu'))
+                'MCEBNfold2.pth', map_location='cpu'))
             model.eval()
             prediction = model(test_images.unsqueeze(1))
             pred = prediction.argmax(dim=1).item()