Hugging Face's logo

Spaces:
fbrynpk
/
Covid-19-Detection
Sleeping

App Files Community
fbrynpk commited on
Commit
cd25bbd
•
1 Parent(s): 650dcdc

Fix misformatting

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -44,7 +44,7 @@ data_transform = transforms.Compose(
44
  )
45
 
46
  EffNetB0_load_model.load_state_dict(
47
- torch.load("./EffNetB0_data_auto_10_epochs.pth"), map_location=torch.device("cpu")
48
  )
49
 
50
  ### Predict function ---------------------------------------------------- ###
 
44
  )
45
 
46
  EffNetB0_load_model.load_state_dict(
47
+ torch.load("./EffNetB0_data_auto_10_epochs.pth", map_location=torch.device("cpu"))
48
  )
49
 
50
  ### Predict function ---------------------------------------------------- ###