Den4ikAI commited on
Commit
a5a9d06
1 Parent(s): 3797a5d

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -2
README.md CHANGED
@@ -28,12 +28,11 @@ def generate(prompt):
28
  **data,
29
  generation_config=generation_config
30
  )[0]
31
- print(tokenizer.decode(data["input_ids"][0].tolist()))
32
  out = tokenizer.decode(output_ids.tolist())
33
  return out
34
 
35
  while 1:
36
- generate(input(":> "))
37
 
38
  ```
39
  # Citation
 
28
  **data,
29
  generation_config=generation_config
30
  )[0]
 
31
  out = tokenizer.decode(output_ids.tolist())
32
  return out
33
 
34
  while 1:
35
+ print(generate(input(":> ")))
36
 
37
  ```
38
  # Citation