add .gitignore, add "if torch.cuda.is_available():" in demo.py

.gitignore

+.DS_Store
+*.pth
+*.pyc
+*.pyo
+*.log
+*.tmp
+

demo.py

@@ -11,7 +11,9 @@ model_path = './data/crnn.pth'
 img_path = './data/demo.png'
 alphabet = '0123456789abcdefghijklmnopqrstuvwxyz'
 
-model = crnn.CRNN(32, 1, 37, 256).cuda()
+model = crnn.CRNN(32, 1, 37, 256)
+if torch.cuda.is_available():
+    model = model.cuda()
 print('loading pretrained model from %s' % model_path)
 model.load_state_dict(torch.load(model_path))
 
@@ -19,7 +21,9 @@ converter = utils.strLabelConverter(alphabet)
 
 transformer = dataset.resizeNormalize((100, 32))
 image = Image.open(img_path).convert('L')
-image = transformer(image).cuda()
+image = transformer(image)
+if torch.cuda.is_available():
+    image = image.cuda()
 image = image.view(1, *image.size())
 image = Variable(image)