Update app.py

app.py

@@ -20,8 +20,8 @@ model_gpt2 = AutoModelForCausalLM.from_pretrained("openai-community/gpt2")
 tokenizer_gemma = AutoTokenizer.from_pretrained("google/gemma-2b")
 model_gemma = AutoModelForCausalLM.from_pretrained("google/gemma-2b", token=token)
 
-tokenizer_gptneo = AutoTokenizer.from_pretrained("EleutherAI/gpt-neo-1.3B")
-model_gptneo = AutoModelForCausalLM.from_pretrained("EleutherAI/gpt-neo-1.3B")
+tokenizer_qwen = AutoTokenizer.from_pretrained("Qwen/Qwen2-0.5B")
+model_qwen = AutoModelForCausalLM.from_pretrained("Qwen/Qwen2-0.5B")
 
 # Define functions
 
@@ -147,9 +147,9 @@ def load_model(model_selected):
         tokenizer = tokenizer_gemma
         model = model_gemma
     
-    if model_selected == "Eleuther GPT Neo 1.3B":
-        tokenizer = tokenizer_gptneo
-        model = model_gptneo
+    if model_selected == "Qwen2":
+        tokenizer = tokenizer_qwen
+        model = model_qwen
 
 
 
@@ -311,7 +311,7 @@ with gr.Blocks() as demo:
         
         with gr.Column (scale=0, min_width=200) as Models_Strategy:
             
-            model_selected = gr.Radio (["GPT2", "Gemma 2", "Eleuther GPT Neo 1.3B"], label="ML Model", value="GPT2")
+            model_selected = gr.Radio (["GPT2", "Gemma 2", "Qwen2"], label="ML Model", value="GPT2")
             strategy_selected = gr.Radio (["Sampling", "Beam Search", "Diversity Beam Search","Contrastive"], label="Search strategy", value = "Sampling", interactive=True)