Update app.py
Browse files
app.py
CHANGED
@@ -94,7 +94,7 @@ def main():
|
|
94 |
|
95 |
|
96 |
print(url_input)
|
97 |
-
cls_model =AutoModelForSequenceClassification.from_pretrained("
|
98 |
tokenizer_cls = AutoTokenizer.from_pretrained("xlnet-base-cased")
|
99 |
label_encoder = LabelEncoder()
|
100 |
|
|
|
94 |
|
95 |
|
96 |
print(url_input)
|
97 |
+
cls_model =AutoModelForSequenceClassification.from_pretrained("riskclassification_finetuned_xlnet_model_ld")
|
98 |
tokenizer_cls = AutoTokenizer.from_pretrained("xlnet-base-cased")
|
99 |
label_encoder = LabelEncoder()
|
100 |
|