bofenghuang
commited on
Commit
•
d460be2
1
Parent(s):
2d54efe
Update README.md
Browse files
README.md
CHANGED
@@ -37,8 +37,7 @@ pipe = pipeline(
|
|
37 |
device=device,
|
38 |
)
|
39 |
|
40 |
-
dialogue_text = """
|
41 |
-
Pierre: J’ai oublié ma trousse. Tu peux me prêter un stylo.
|
42 |
Lucie: Tiens.
|
43 |
Pierre: Merci. Tu peux me donner une feuille de papier aussi ?
|
44 |
Lucie: Euh… oui. Tiens.
|
@@ -49,8 +48,7 @@ Lucie: Oui, si tu veux… On ira au café.
|
|
49 |
Pierre: Oui… euh non, j’ai oublié mon porte-monnaie.
|
50 |
Lucie: Bon allez ! ce n’est pas grave, je t’invite.
|
51 |
Pierre: Tu es trop gentille.
|
52 |
-
Lucie: Oui, c’est bien possible.
|
53 |
-
"""
|
54 |
|
55 |
summarized_text = pipe(dialogue_text, max_length=1024)[0]["summary_text"] # greedy
|
56 |
# summarized_text = pipe(dialogue_text, max_length=1024, num_beams=5)[0]["summary_text"] # beam search
|
|
|
37 |
device=device,
|
38 |
)
|
39 |
|
40 |
+
dialogue_text = """Pierre: J’ai oublié ma trousse. Tu peux me prêter un stylo.
|
|
|
41 |
Lucie: Tiens.
|
42 |
Pierre: Merci. Tu peux me donner une feuille de papier aussi ?
|
43 |
Lucie: Euh… oui. Tiens.
|
|
|
48 |
Pierre: Oui… euh non, j’ai oublié mon porte-monnaie.
|
49 |
Lucie: Bon allez ! ce n’est pas grave, je t’invite.
|
50 |
Pierre: Tu es trop gentille.
|
51 |
+
Lucie: Oui, c’est bien possible."""
|
|
|
52 |
|
53 |
summarized_text = pipe(dialogue_text, max_length=1024)[0]["summary_text"] # greedy
|
54 |
# summarized_text = pipe(dialogue_text, max_length=1024, num_beams=5)[0]["summary_text"] # beam search
|