Spaces:

Thenujan
/

Covid_CNN

Runtime error

Thenujan commited on Oct 5, 2023

Commit

ba9fd81

1 Parent(s): 4d3c006

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -36,7 +36,7 @@ elif model_name == 'resnet':
     # Load saved weights
     model.load_state_dict(
         torch.load(
-            f="./cnn-custom-model-version-5.pt",
             map_location=torch.device("cpu"),  # load to CPU
         )
     )

     # Load saved weights
     model.load_state_dict(
         torch.load(
+            f="./cnn-custom-model-version-2.pt",
             map_location=torch.device("cpu"),  # load to CPU
         )
     )