Update app.py

app.py

@@ -14,8 +14,8 @@ token = os.environ.get("HF_TOKEN")
 #tokenizer = AutoTokenizer.from_pretrained("openai-community/gpt2")
 #model = AutoModelForCausalLM.from_pretrained("openai-community/gpt2")
 
-tokenizer_gpt2 = AutoTokenizer.from_pretrained("openai-community/gpt2")
-model_gpt2 = AutoModelForCausalLM.from_pretrained("openai-community/gpt2")
+#tokenizer_gpt2 = AutoTokenizer.from_pretrained("openai-community/gpt2")
+#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)
@@ -138,9 +138,9 @@ def generate(input_text, number_steps, number_beams, number_beam_groups, diversi
 
 def load_model(model_selected):
     
-    if model_selected == "GPT2":
-        tokenizer = tokenizer_gpt2
-        model = model_gpt2
+    #if model_selected == "GPT2":
+        #tokenizer = tokenizer_gpt2
+        #model = model_gpt2
         #print (model_selected + " loaded")
     
     #if model_selected == "Gemma 2":
@@ -304,8 +304,8 @@ with gr.Blocks() as demo:
     
     No_beam_group_list = [2]
         
-    tokenizer = tokenizer_gpt2
-    model = model_gpt2
+    tokenizer = tokenizer_qwen
+    model = model_qwen
     
     with gr.Row():