Spaces:

madhurjindal
/

image_classification_cifar10_gradcam

Sleeping

madhurjindal commited on Aug 5, 2023

Commit

b923f28

•

1 Parent(s): b54de11

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -17,8 +17,8 @@ model = Net(
     weight_decay=config.WEIGHT_DECAY,
 )
-model.load_state_dict(torch.load(config.MODEL_SAVE_PATH))
-model.pred_store = torch.load(config.PRED_STORE_PATH)
 def generate_gradio_output(

     weight_decay=config.WEIGHT_DECAY,
 )
+model.load_state_dict(torch.load(config.MODEL_SAVE_PATH, map_location=torch.device('cpu')))
+model.pred_store = torch.load(config.PRED_STORE_PATH, map_location=torch.device('cpu'))
 def generate_gradio_output(