Spaces:
Runtime error
Runtime error
Update generic.py
Browse files- generic.py +2 -2
generic.py
CHANGED
@@ -13,9 +13,9 @@ import torch
|
|
13 |
|
14 |
from sequenceoutput.modeling_output import SequenceClassifierOutput
|
15 |
|
16 |
-
model = BertForSequenceClassification.from_pretrained("BERT_weight")
|
17 |
model.eval()
|
18 |
-
tokenizer = AutoTokenizer.from_pretrained("BERT_weight")
|
19 |
# initialize the explanations generator
|
20 |
explanations = Generator(model)
|
21 |
|
|
|
13 |
|
14 |
from sequenceoutput.modeling_output import SequenceClassifierOutput
|
15 |
|
16 |
+
model = BertForSequenceClassification.from_pretrained("./BERT/BERT_weight")
|
17 |
model.eval()
|
18 |
+
tokenizer = AutoTokenizer.from_pretrained("./BERT/BERT_weight")
|
19 |
# initialize the explanations generator
|
20 |
explanations = Generator(model)
|
21 |
|