Update app.py

app.py

@@ -37,9 +37,9 @@ css = """
     height: 62px;
 }
 """
-device = 'cuda' if torch.cuda.is_available() else 'cpu'
+# device = 'cuda' if torch.cuda.is_available() else 'cpu'
 
-# device = torch.device('cpu')
+device = torch.device('cpu')
 
 parser = argparse.ArgumentParser('PDFNet Testing script', parents=[get_args_parser()])
 args = parser.parse_args(args=[])