Frederick commited on
Commit
8086f1f
·
1 Parent(s): 053f51f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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("C:\\Users\\pixin\\Desktop\\Reddit ML\\Trained Models\\clause_model_512", num_labels=3)
22
- classification_model = RobertaForSequenceClassification.from_pretrained("C:\\Users\\pixin\\Desktop\\Reddit ML\\Trained Models\\classfication_model", num_labels=18)
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 = {