lorenzoscottb commited on
Commit
af9ef3e
·
1 Parent(s): 0b8554b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -0
app.py CHANGED
@@ -3,7 +3,13 @@ import pickle
3
  import pandas as pd
4
 
5
  data = pickle.load(open("OOV_Train_2.pkl", "rb"))
 
 
 
 
6
 
 
 
7
 
8
  description = "Did you know that, logically speaking, a small cat is not a small animal, and that a fake smile is not a smile. \n Learn more testing our BERT model is tuned to perform phrase-level adjective-noun entailment, via the [PLANE](https://aclanthology.org/2022.coling-1.359/) dataset. \n The current model achieves an accuracy of 90% on out-of-distribution evaluation"
9
  title = "BERT on a PLANE"
 
3
  import pandas as pd
4
 
5
  data = pickle.load(open("OOV_Train_2.pkl", "rb"))
6
+ data = pd.DataFrame(
7
+ data,
8
+ columns=["Input_Seq", "Label", "Adj_Class", "Adj", "Nn", "Hypr", "Adj_NN"]
9
+ )
10
 
11
+ adjs = set(data["Adj"])
12
+ Nns = set(list(data["Nn"]) + list(data["Hypr"]))
13
 
14
  description = "Did you know that, logically speaking, a small cat is not a small animal, and that a fake smile is not a smile. \n Learn more testing our BERT model is tuned to perform phrase-level adjective-noun entailment, via the [PLANE](https://aclanthology.org/2022.coling-1.359/) dataset. \n The current model achieves an accuracy of 90% on out-of-distribution evaluation"
15
  title = "BERT on a PLANE"