Spaces:

3koozy
/

FloodNet_AAM_Team

Runtime error

3koozy commited on Mar 30, 2023

Commit

4076aa5

•

1 Parent(s): e1d6224

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -9,7 +9,7 @@ model = smp.Unet(
     in_channels=3,                  # model input channels (1 for gray-scale images, 3 for RGB, etc.)
     classes=10,                      # model output channels (number of classes in your dataset)
 )
-model.load_state_dict(torch.load('Floodnet_model_e5.pt'))
 model.eval()
 #handle input:

     in_channels=3,                  # model input channels (1 for gray-scale images, 3 for RGB, etc.)
     classes=10,                      # model output channels (number of classes in your dataset)
 )
+model.load_state_dict(torch.load('Floodnet_model_e5.pt', map_location=torch.device('cpu')))
 model.eval()
 #handle input: