Update app.py
Browse files
app.py
CHANGED
|
@@ -18,8 +18,8 @@ import gradio as gr
|
|
| 18 |
os.system("python -m spacy download en_core_web_sm")
|
| 19 |
nlp = spacy.load("en_core_web_sm")
|
| 20 |
tokenizer = RobertaTokenizer.from_pretrained("roberta-base")
|
| 21 |
-
clause_model = AutoModelForTokenClassification.from_pretrained("
|
| 22 |
-
classification_model = RobertaForSequenceClassification.from_pretrained("
|
| 23 |
|
| 24 |
|
| 25 |
labels2attrs = {
|
|
|
|
| 18 |
os.system("python -m spacy download en_core_web_sm")
|
| 19 |
nlp = spacy.load("en_core_web_sm")
|
| 20 |
tokenizer = RobertaTokenizer.from_pretrained("roberta-base")
|
| 21 |
+
clause_model = AutoModelForTokenClassification.from_pretrained("./clause_model_512", num_labels=3)
|
| 22 |
+
classification_model = RobertaForSequenceClassification.from_pretrained("./classfication_model", num_labels=18)
|
| 23 |
|
| 24 |
|
| 25 |
labels2attrs = {
|