commit

run_segbot.py

@@ -66,8 +66,8 @@ def setup():
     with open('index2word.pickle', 'rb') as f:    
         index2word = pickle.load(f)
     with open('model.pickle', 'rb') as f:
-        mysolver = pickle.load(f)
+        mysolver = torch.load(f, torch.device('cpu'))
     with open('fm.pickle', 'rb') as f:
         fm = pickle.load(f)
     
-    return mysolver,fm,index2word
+    return mysolver,fm,index2word