Spaces:
Sleeping
Sleeping
Changed to CPU device
Browse files
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)
|