loubnabnl HF staff commited on
Commit
53b0b9d
1 Parent(s): 0847ae0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -10,8 +10,8 @@ example = [
10
  ["def get_file_size(filepath):", 40, 0.6, 42],
11
  ["def count_lines(filename):", 40, 0.6, 42],
12
  ["def count_words(filename):", 40, 0.6, 42]]
13
- tokenizer = AutoTokenizer.from_pretrained("lvwerra/codeparrot")
14
- model = AutoModelForCausalLM.from_pretrained("lvwerra/codeparrot", low_cpu_mem_usage=True)
15
 
16
 
17
  def code_generation(gen_prompt, max_tokens, temperature=0.6, seed=42):
 
10
  ["def get_file_size(filepath):", 40, 0.6, 42],
11
  ["def count_lines(filename):", 40, 0.6, 42],
12
  ["def count_words(filename):", 40, 0.6, 42]]
13
+ tokenizer = AutoTokenizer.from_pretrained("codeparrot/codeparrot")
14
+ model = AutoModelForCausalLM.from_pretrained("codeparrot/codeparrot", low_cpu_mem_usage=True)
15
 
16
 
17
  def code_generation(gen_prompt, max_tokens, temperature=0.6, seed=42):