hotfix

app.py

@@ -1,7 +1,8 @@
 import gradio as gr
 from NeuralTextGenerator import BertTextGenerator
 
-model_name = "cardiffnlp/twitter-xlm-roberta-base" #"dbmdz/bert-base-italian-uncased"
+# Load models
+model_name = "cardiffnlp/twitter-xlm-roberta-base"
 en_model = BertTextGenerator(model_name, tokenizer=model_name)
 
 finetunned_BERT_model_name = "JuanJoseMV/BERT_text_gen"
@@ -19,14 +20,14 @@ special_tokens = [
     '[NEGATIVE-2]'
     ]
 
-en_model.tokenizer.add_special_tokens({'additional_special_tokens': special_tokens})
-en_model.model.resize_token_embeddings(len(en_model.tokenizer))
+# en_model.tokenizer.add_special_tokens({'additional_special_tokens': special_tokens})
+# en_model.model.resize_token_embeddings(len(en_model.tokenizer))
 
 finetunned_BERT_en_model.tokenizer.add_special_tokens({'additional_special_tokens': special_tokens})
 finetunned_BERT_en_model.model.resize_token_embeddings(len(en_model.tokenizer))
 
-finetunned_RoBERTa_en_model.tokenizer.add_special_tokens({'additional_special_tokens': special_tokens})
-finetunned_RoBERTa_en_model.model.resize_token_embeddings(len(en_model.tokenizer))
+# finetunned_RoBERTa_en_model.tokenizer.add_special_tokens({'additional_special_tokens': special_tokens})
+# finetunned_RoBERTa_en_model.model.resize_token_embeddings(len(en_model.tokenizer))
 
 def sentence_builder(selected_model, n_sentences, max_iter, sentiment, seed_text):