Hugging Face's logo

Spaces:
guychuk
/
codegen-demo
Sleeping

App Files Community
guychuk commited on
Commit
14a8fa8
·
verified ·
1 Parent(s): 361b027

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +0 -1
app.py CHANGED
@@ -11,7 +11,6 @@ tokenizer = AutoTokenizer.from_pretrained(model_name)
11
  model = AutoModelForCausalLM.from_pretrained(model_name)
12
 
13
  model.to(device)
14
- tokenizer.to(device)
15
 
16
  def generate_code(prompt):
17
  full_prompt = f"# HTML/CSS/JS website code for:\n# {prompt}\n"
 
11
  model = AutoModelForCausalLM.from_pretrained(model_name)
12
 
13
  model.to(device)
 
14
 
15
  def generate_code(prompt):
16
  full_prompt = f"# HTML/CSS/JS website code for:\n# {prompt}\n"