Update app.py

app.py

@@ -131,12 +131,26 @@ def generate(input_text, number_steps, number_beams, number_beam_groups, diversi
     
 #--------ON SELECTING MODEL------------------------
 
+def select_model(model_selected):
+    
+    global model_name
+    
+    if model_selected == "GPT2":
+        model_name = "openai-community/gpt2"
+        #tokenizer = tokenizer_gpt2
+        #model = model_gpt2
+        #print (model_selected + " loaded")
+    
+    if model_selected == "Qwen2":
+        model_name = "Qwen/Qwen2-0.5B"
+        #tokenizer = tokenizer_qwen
+        #model = model_qwen
 
 # On clicking load button
 
 def load_model (name):
     
-    #global model_name
+    global model_name
     global tokenizer
     global model