temp

app.py

@@ -7,6 +7,8 @@ import fire_network
 # Possible Scales for multiscale inference
 scales = [2.0, 1.414, 1.0, 0.707, 0.5, 0.353, 0.25] 
 
+device = 'cpu'
+
 # Load net
 state = torch.load('fire.pth', map_location='cpu')
 state['net_params']['pretrained'] = None # no need for imagenet pretrained model