Update app.py

app.py

@@ -160,14 +160,14 @@ def select_model(model_selected):
 
 # On clicking load button
 
-def load_model ():
+def load_model (name):
     
     global model_name
     global tokenizer
     global model
     
-    tokenizer = AutoTokenizer.from_pretrained(model_name)
-    model = AutoModelForCausalLM.from_pretrained(model_name)
+    tokenizer = AutoTokenizer.from_pretrained(name)
+    model = AutoModelForCausalLM.from_pretrained(name)
 
     
 
@@ -329,7 +329,7 @@ with gr.Blocks() as demo:
         
         with gr.Column (scale=0, min_width=200) as Models_Strategy:
             
-            model_selected = gr.Radio (["GPT2", "Qwen2"], label="ML Model", value="GPT2")
+            model_selected = gr.Radio (["GPT2", "Qwen/Qwen2-0.5B"], label="ML Model", value="GPT2")
             load_model_button = gr.Button("Load")
             strategy_selected = gr.Radio (["Sampling", "Beam Search", "Diversity Beam Search","Contrastive"], label="Search strategy", value = "Sampling", interactive=True)