RamAI123 commited on
Commit
202989b
1 Parent(s): 4914872

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -94,7 +94,7 @@ def main():
94
 
95
 
96
  print(url_input)
97
- cls_model =AutoModelForSequenceClassification.from_pretrained("/content/drive/MyDrive/riskclassification_finetuned_xlnet_model_ld")
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 = "/content/drive/MyDrive/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")
 
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")