numb3r3 commited on
Commit
93b98c1
·
verified ·
1 Parent(s): 352f47d

fix error in the html-to-json example

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -211,7 +211,7 @@ schema = """
211
  """
212
 
213
  html = clean_html(html)
214
- input_prompt = create_prompt(html, schema=schema)
215
 
216
  inputs = tokenizer.encode(input_prompt, return_tensors="pt").to(device)
217
  outputs = model.generate(
 
211
  """
212
 
213
  html = clean_html(html)
214
+ input_prompt = create_prompt(html, tokenizer=tokenizer, schema=schema)
215
 
216
  inputs = tokenizer.encode(input_prompt, return_tensors="pt").to(device)
217
  outputs = model.generate(