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 |
|
@@ -109,7 +109,7 @@ def main():
|
|
109 |
print("Label encoder values")
|
110 |
|
111 |
# Replace the original column with the encoded values
|
112 |
-
label_encoder_path = "
|
113 |
joblib.dump(label_encoder, label_encoder_path)
|
114 |
|
115 |
model_summ = T5ForConditionalGeneration.from_pretrained("t5-small")
|
|
|
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 |
|
|
|
109 |
print("Label encoder values")
|
110 |
|
111 |
# Replace the original column with the encoded values
|
112 |
+
label_encoder_path = "riskclassification_finetuned_xlnet_model_ld/encoder_labels.pkl"
|
113 |
joblib.dump(label_encoder, label_encoder_path)
|
114 |
|
115 |
model_summ = T5ForConditionalGeneration.from_pretrained("t5-small")
|