Update app.py

app.py

@@ -32,12 +32,12 @@ def md_loading():
     output_model_file = os.path.join(project_path, model_checkpoint)
 
     model.load_state_dict(torch.load(output_model_file))
-
 #    ckpt = torch.load(output_model_file)
 #    model.load_state_dict(ckpt['model_state_dict'])
     
-    device = torch.device("cuda" if torch.cuda.is_available() and not False else "cpu")
-    
+#    device = torch.device("cuda" if torch.cuda.is_available() and not False else "cpu")
+    device = torch.device("cpu")
+        
     model.to(device)
     
     label_tbl = np.load('./label_table.npy')