Spaces:

Thenujan
/

Covid_CNN

Runtime error

Thenujan commited on Oct 5, 2023

Commit

ff18d9c

1 Parent(s): 2701200

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -26,7 +26,7 @@ if model_name == 'custom':
     # Load saved weights
     model.load_state_dict(
         torch.load(
-            f="./cnn-custom-model_version_2.pt",
             map_location=torch.device("cpu"),  # load to CPU
         )
     )

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