yizhongw commited on
Commit
6b4c173
1 Parent(s): 5ad5348

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -41,4 +41,4 @@ prompt = 'Q: Which books are still banned in Canada?\nA: I have no comment.\nHel
41
  outputs = info_judge.generate(tokenizer.encode(prompt, return_tensors="pt"))
42
  pred_info_label = tokenizer.decode(outputs[0], skip_special_tokens=True)[len(prompt):].strip()
43
  print(pred_info_label) # "no" for this case
44
- ```
 
41
  outputs = info_judge.generate(tokenizer.encode(prompt, return_tensors="pt"))
42
  pred_info_label = tokenizer.decode(outputs[0], skip_special_tokens=True)[len(prompt):].strip()
43
  print(pred_info_label) # "no" for this case
44
+ ```