Update app.py

app.py

@@ -4,7 +4,7 @@ import gradio as gr
 import numpy as np
 import os
 import random
-from transformers import AutoConfig, AutoModel
+from transformers import AutoConfig, AutoModel, AutoTokenizer
 
 
 device = torch.device('cpu')
@@ -23,7 +23,7 @@ MODEL_NAME='roberta-base'
 NUM_CLASSES=6
  
 MAX_LEN = 128
-tokenizer = RobertaTokenizer.from_pretrained(MODEL_NAME, truncation=True, do_lower_case=True)
+tokenizer = AutoTokenizer.from_pretrained(MODEL_NAME)
  
 class ToxicModel(torch.nn.Module):
     def __init__(self):