Update main.py

main.py

@@ -1,7 +1,7 @@
 import streamlit as st
 import json
 import torch
-from transformers import AutoTokenizer
+from transformers import AutoTokenizer, AutoModelForTokenClassification
 from modelling_cnn import CNNForNER, SentimentCNNModel
 import pandas as pd
 import altair as alt
@@ -21,7 +21,7 @@ import altair as alt
 # ner_model.eval()
 
 ner_model = AutoModelForTokenClassification.from_pretrained("masakhane/afroxlmr-large-ner-masakhaner-1.0_2.0")
-ner_tokenizers = AutoTokenizer.from_pretrained("masakhane/afroxlmr-large-ner-masakhaner-1.0_2.0")
+ner_tokenizer = AutoTokenizer.from_pretrained("masakhane/afroxlmr-large-ner-masakhaner-1.0_2.0")
 ner_config = ner_model.config
 
 ner_model.eval()