Spaces:
Running
Running
Saved new jit model
Browse files- app.py +1 -1
- final_model.pth → final_model.pt +2 -2
app.py
CHANGED
@@ -11,7 +11,7 @@ import re
|
|
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.
|
15 |
correct_preds, wrong_preds = {}, {}
|
16 |
condition_lst = pd.read_csv("feature.csv", header = "infer", sep = ",", encoding = "utf-8", dtype=str)
|
17 |
D_LABITEMS = pd.read_csv("D_LABITEMS.csv", header = "infer", sep = ",", encoding = "utf-8", dtype=str)
|
|
|
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.pt").to(device)
|
15 |
correct_preds, wrong_preds = {}, {}
|
16 |
condition_lst = pd.read_csv("feature.csv", header = "infer", sep = ",", encoding = "utf-8", dtype=str)
|
17 |
D_LABITEMS = pd.read_csv("D_LABITEMS.csv", header = "infer", sep = ",", encoding = "utf-8", dtype=str)
|
final_model.pth → final_model.pt
RENAMED
@@ -1,3 +1,3 @@
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
-
oid sha256:
|
3 |
-
size
|
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:0aca72505f3e02d8c071fa4cf61de1d1f976165570d34590770952aa96736221
|
3 |
+
size 60999015
|