Spaces:

fbrynpk
/

pneumonia-detection

Build error

fbrynpk commited on Oct 25, 2023

Commit

650dcdc

1 Parent(s): d3c48d3

Set default device to cpu

Files changed (1) hide show

app.py CHANGED Viewed

@@ -25,7 +25,7 @@ EffNetB0_load_model, EffNetB0_transforms = create_effnet(
     in_features=1280,
     dropout=0.2,
     out_features=len(class_name),
-    device=device,
 )
 # Write a transform for image

     in_features=1280,
     dropout=0.2,
     out_features=len(class_name),
+    device="cpu",
 )
 # Write a transform for image