hotfix

app.py

@@ -2,13 +2,13 @@ import gradio as gr
 from NeuralTextGenerator import BertTextGenerator
 
 model_name = "cardiffnlp/twitter-xlm-roberta-base" #"dbmdz/bert-base-italian-uncased"
-en_model = BertTextGenerator(model_name, tokenizer='xlm-roberta')
+en_model = BertTextGenerator(model_name, tokenizer=model_name)
 
 finetunned_BERT_model_name = "JuanJoseMV/BERT_text_gen"
 finetunned_BERT_en_model = BertTextGenerator(finetunned_BERT_model_name, tokenizer='bert-base-uncased')
 
 finetunned_RoBERTa_model_name = "JuanJoseMV/XLM_RoBERTa_text_gen"
-finetunned_RoBERTa_en_model = BertTextGenerator(finetunned_RoBERTa_model_name, tokenizer='xlm-roberta')
+finetunned_RoBERTa_en_model = BertTextGenerator(finetunned_RoBERTa_model_name, tokenizer=finetunned_RoBERTa_model_name)
 
 special_tokens = [
     '[POSITIVE-0]',