Spaces:

codeparrot
/

codeparrot-subspace

Runtime error

loubnabnl HF Staff commited on May 24, 2022

Commit

31b8e2c

1 Parent(s): 416adec

update app

Files changed (1) hide show

app.py CHANGED Viewed

@@ -14,7 +14,6 @@ model = AutoModelForCausalLM.from_pretrained("lvwerra/codeparrot", low_cpu_mem_u
 def code_generation(gen_prompt, max_tokens, temperature=0.6, seed=42):
     set_seed(seed)
-    gen_kwargs = {}
     pipe = pipeline("text-generation", model=model, tokenizer=tokenizer)
     generated_text = pipe(gen_prompt, do_sample=True, top_p=0.95, temperature=temperature, max_new_tokens=max_tokens)[0]['generated_text']
     return generated_text

 def code_generation(gen_prompt, max_tokens, temperature=0.6, seed=42):
     set_seed(seed)
     pipe = pipeline("text-generation", model=model, tokenizer=tokenizer)
     generated_text = pipe(gen_prompt, do_sample=True, top_p=0.95, temperature=temperature, max_new_tokens=max_tokens)[0]['generated_text']
     return generated_text