bofenghuang commited on
Commit
d460be2
1 Parent(s): 2d54efe

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -4
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