tnt306 commited on
Commit
3452cb6
·
1 Parent(s): 7514cfe

Saved new jit model

Browse files
Files changed (2) hide show
  1. app.py +1 -1
  2. 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.pth").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)
 
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:731461d450bb0ec5f011c3751c98621171fb9abd017331da48d1f6fb3dec184d
3
- size 61002953
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:0aca72505f3e02d8c071fa4cf61de1d1f976165570d34590770952aa96736221
3
+ size 60999015