Update app.py

app.py

@@ -8,20 +8,12 @@ import os
 
 token = os.environ.get("HF_TOKEN")
 
-# Load default model as GPT2 and other models
+# Load default model as GPT2
 
 
-#tokenizer = AutoTokenizer.from_pretrained("openai-community/gpt2")
-#model = AutoModelForCausalLM.from_pretrained("openai-community/gpt2")
+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_gemma = AutoTokenizer.from_pretrained("google/gemma-2b")
-#model_gemma = AutoModelForCausalLM.from_pretrained("google/gemma-2b", token=token)
-
-#tokenizer_qwen = AutoTokenizer.from_pretrained("Qwen/Qwen2-0.5B")
-#model_qwen = AutoModelForCausalLM.from_pretrained("Qwen/Qwen2-0.5B")
 
 # Define functions
 
@@ -139,30 +131,12 @@ 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 == "Gemma 2":
-        #tokenizer = tokenizer_gemma
-        #model = model_gemma
-    
-    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