Commit
•
ee1cea2
1
Parent(s):
c29e4cf
Update README.md
Browse files
README.md
CHANGED
@@ -71,7 +71,7 @@ Daniel: I really hope Amiens is worth it!
|
|
71 |
"""
|
72 |
|
73 |
input = tokenizer(text, return_tensors="pt")
|
74 |
-
outputs = model.generate(input_ids=input["input_ids"].cuda(), max_new_tokens=40)
|
75 |
print("Summary: ", tokenizer.decode(outputs[0], skip_special_tokens=True))
|
76 |
|
77 |
Summary: Emma and Peter are going to Amiens tomorrow. Daniel will join them. They will meet at Emma's place by 9 am. They will not have any parties tonight.
|
|
|
71 |
"""
|
72 |
|
73 |
input = tokenizer(text, return_tensors="pt")
|
74 |
+
outputs = model.generate(input_ids=input["input_ids"].cuda(), max_new_tokens=40)
|
75 |
print("Summary: ", tokenizer.decode(outputs[0], skip_special_tokens=True))
|
76 |
|
77 |
Summary: Emma and Peter are going to Amiens tomorrow. Daniel will join them. They will meet at Emma's place by 9 am. They will not have any parties tonight.
|