Update README.md
Browse files
README.md
CHANGED
@@ -51,9 +51,16 @@ sample_outputs = model.generate(
|
|
51 |
num_return_sequences=20
|
52 |
)
|
53 |
|
54 |
-
print("Output:\
|
|
|
55 |
for i, sample_output in enumerate(sample_outputs):
|
56 |
m = tokenizer.decode(sample_output, skip_special_tokens=True)
|
57 |
m = m.split("TODO")[1].strip()
|
58 |
print("{}: TODO{}".format(i, m))
|
59 |
```
|
|
|
|
|
|
|
|
|
|
|
|
|
|
51 |
num_return_sequences=20
|
52 |
)
|
53 |
|
54 |
+
print("Output:\
|
55 |
+
" + 100 * '-')
|
56 |
for i, sample_output in enumerate(sample_outputs):
|
57 |
m = tokenizer.decode(sample_output, skip_special_tokens=True)
|
58 |
m = m.split("TODO")[1].strip()
|
59 |
print("{}: TODO{}".format(i, m))
|
60 |
```
|
61 |
+
|
62 |
+
|
63 |
+
## TODO
|
64 |
+
|
65 |
+
- [ ] Preprocess the data in a better way
|
66 |
+
- [ ] Download github and train it on everything
|