Hugging Face's logo

Spaces:
Thenujan
/
Covid_CNN
Runtime error

App Files Community
Thenujan commited on
Commit
a2d67f9
·
1 Parent(s): 85bc780

Added models

Browse files
Files changed (4) hide show
  1. app.py +1 -1
  2. cnn-custom-model-version-3.pt +3 -0
  3. model_cnn_proj_version_1.pt → cnn-custom-model_version_1.pt +0 -0
  4. model_cnn_proj_version_2.pt → cnn-custom-model_version_2.pt +0 -0
app.py CHANGED
@@ -23,7 +23,7 @@ model = create_custom_model()
23
  # Load saved weights
24
  model.load_state_dict(
25
  torch.load(
26
- f="./model_cnn_proj_version_2.pt",
27
  map_location=torch.device("cpu"), # load to CPU
28
  )
29
  )
 
23
  # Load saved weights
24
  model.load_state_dict(
25
  torch.load(
26
+ f="./cnn-custom-model-version-3.pt",
27
  map_location=torch.device("cpu"), # load to CPU
28
  )
29
  )
cnn-custom-model-version-3.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:20517dac4c105c441790ef2eb92cd229a206d89653ff3a2a7db51384e442150e
3
+ size 179543
model_cnn_proj_version_1.pt → cnn-custom-model_version_1.pt RENAMED
File without changes
model_cnn_proj_version_2.pt → cnn-custom-model_version_2.pt RENAMED
File without changes