Update app.py

app.py

@@ -129,8 +129,8 @@ def generate(input_text, number_steps, number_beams, number_beam_groups, diversi
 def load_model(model_selected):
     
     if model_selected == "gpt2":
-        tokenizer = AutoTokenizer.from_pretrained("gpt2")
-        model = AutoModelForCausalLM.from_pretrained("gpt2", pad_token_id=tokenizer.eos_token_id)
+        tokenizer = AutoTokenizer.from_pretrained("openai-community/gpt2")
+        model = AutoModelForCausalLM.from_pretrained("openai-community/gpt2", pad_token_id=tokenizer.eos_token_id)
         #print (model_selected + " loaded")
     
     if model_selected == "Gemma 2":
@@ -294,8 +294,8 @@ with gr.Blocks() as demo:
         value="It's a rainy day today",
     )
     
-    tokenizer = AutoTokenizer.from_pretrained("gpt2")
-    model = AutoModelForCausalLM.from_pretrained("gpt2", pad_token_id=tokenizer.eos_token_id, cache_dir=cache_dir).to(torch_device)
+    tokenizer = AutoTokenizer.from_pretrained("openai-community/gpt2")
+    model = AutoModelForCausalLM.from_pretrained("openai-community/gpt2", pad_token_id=tokenizer.eos_token_id)
 
     
     with gr.Row():