Spaces:

mirari
/

Practica2

Runtime error

mirari commited on May 30, 2022

Commit

620ffb5

1 Parent(s): 83bcea0

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -1,7 +1,7 @@
 from icevision.all import *
 import gradio as gr
-state_dict = torch.load('fasterRCNNKangaroo.pth')
 model.load_state_dict(state_dict)
 labels = learn.dls.vocab

 from icevision.all import *
 import gradio as gr
+state_dict = torch.load('fasterRCNNKangaroo.pth', map_location=torch.device('cpu'))
 model.load_state_dict(state_dict)
 labels = learn.dls.vocab