fix

app.py

@@ -11,10 +11,10 @@ def load_model(tokenizer_name):
     model_name_dict = {
         "BERT":"bert-base-uncased",
         "RoBERTa":"roberta-base",
-        "ALBERT":"albert-v2-base",
+        "ALBERT":"albert-base-v2",
         "GPT2":"gpt2",
-        "Llama":"meta-lama/Llama-2-7b-chat-hf",
-        "Gemma":"google/gemma-7b",
+        #"Llama":"meta-lama/Llama-2-7b-chat-hf",
+        #"Gemma":"google/gemma-7b",
         }
     tokenizer = AutoTokenizer.from_pretrained(model_name_dict[tokenizer_name])
     return tokenizer
@@ -100,7 +100,7 @@ if __name__=='__main__':
     st.sidebar.write('1. Choose the tokenizer from below')
     tokenizer_name = st.sidebar.selectbox('',
                                             ("BERT","RoBERTa","ALBERT",
-                                             "GPT2","Llama","Gemma"))
+                                             "GPT2"))
     tokenizer = load_model(tokenizer_name)
 
     st.sidebar.write('2. Optional settings')