tnt306 commited on
Commit
7514cfe
·
1 Parent(s): 289248b

Changed to CPU device

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -9,7 +9,8 @@ import re
9
 
10
 
11
  examples_path = "examples"
12
- device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 
13
  model = torch.jit.load("final_model.pth").to(device)
14
  correct_preds, wrong_preds = {}, {}
15
  condition_lst = pd.read_csv("feature.csv", header = "infer", sep = ",", encoding = "utf-8", dtype=str)
 
9
 
10
 
11
  examples_path = "examples"
12
+ #device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
13
+ device = torch.device("cpu")
14
  model = torch.jit.load("final_model.pth").to(device)
15
  correct_preds, wrong_preds = {}, {}
16
  condition_lst = pd.read_csv("feature.csv", header = "infer", sep = ",", encoding = "utf-8", dtype=str)