Spaces:

Seetha
/

IMA-pipeline-streamlit

Sleeping

Seetha commited on Jun 1, 2023

Commit

5106269

•

1 Parent(s): 21f0416

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -115,7 +115,7 @@ def main():
         if lab['label'] == 'causal': #causal
             causal_sents.append(sent)
-  model_name = "distilbert-base-uncased"
   tokenizer = DistilBertTokenizerFast.from_pretrained(model_name)
   model_path1 = "DistilBertforTokenclassification"

         if lab['label'] == 'causal': #causal
             causal_sents.append(sent)
+  model_name = "distilbert-base-cased"
   tokenizer = DistilBertTokenizerFast.from_pretrained(model_name)
   model_path1 = "DistilBertforTokenclassification"