Spaces:

vonewman
/

ner_app

Runtime error

vonewman commited on Nov 3, 2023

Commit

345d423

•

1 Parent(s): 92910b2

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -21,9 +21,9 @@ def convert_json(df: pd.DataFrame):
     return json_string
 def load_model():
-    model = AutoModelForTokenClassification.from_pretrained("vonewman/xlm-roberta-base-finetuned-wolof")
     trainer = Trainer(model=model)
-    tokenizer = AutoTokenizer.from_pretrained("vonewman/xlm-roberta-base-finetuned-wolof")
     return trainer, model, tokenizer
 def align_word_ids(texts):

     return json_string
 def load_model():
+    model = AutoModelForTokenClassification.from_pretrained("vonewman/wolof-finetuned-ner")
     trainer = Trainer(model=model)
+    tokenizer = AutoTokenizer.from_pretrained("vonewman/wolof-finetuned-ner")
     return trainer, model, tokenizer
 def align_word_ids(texts):