fix buf in CPU mode

Author: YoungminBaek

test.py

@@ -109,7 +109,10 @@ def test_net(net, image, text_threshold, link_threshold, low_text, cuda):
     net = CRAFT()     # initialize
 
     print('Loading weights from checkpoint (' + args.trained_model + ')')
-    net.load_state_dict(copyStateDict(torch.load(args.trained_model)))
+    if args.cuda:
+        net.load_state_dict(copyStateDict(torch.load(args.trained_model)))
+    else:
+        net.load_state_dict(copyStateDict(torch.load(args.trained_model, map_location='cpu')))
 
     if args.cuda:
         net = net.cuda()