change network to gpt2

app.py

@@ -11,9 +11,9 @@ example = [
     ["def get_file_size(filepath):", 24, 0.6, 42],
     ["def count_lines(filename):", 40, 0.6, 42],
     ["def count_words(filename):", 40, 0.6, 42]]
-tokenizer = AutoTokenizer.from_pretrained("lvwerra/codeparrot")
-model = AutoModelForCausalLM.from_pretrained("lvwerra/codeparrot", low_cpu_mem_usage=True)
 
+tokenizer = AutoTokenizer.from_pretrained("gpt2")
+model = AutoModelForCausalLM.from_pretrained("gpt2", low_cpu_mem_usage=True)
 
 def code_generation(gen_prompt, max_tokens, temperature=0.6, seed=42):
     set_seed(seed)