File size: 2,368 Bytes
18092fc
 
 
 
 
 
 
a732dff
643d5ea
18092fc
 
 
 
 
 
 
 
 
 
 
 
 
887c2c4
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
57660cf
 
 
 
887c2c4
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
---
license: apache-2.0
language: fr
library_name: transformers
thumbnail: null
tags:
- summarization
widget:
- text: "Pierre: J’ai oublié ma trousse. Tu peux me prêter un stylo.\nLucie: Tiens.\nPierre: Merci. Tu peux me donner une feuille de papier aussi ?\nLucie: Euh… oui. Tiens.\nPierre: Merci. Ça t’ennuie pas si je regarde avec toi ? J’ai oublié mon livre…\nLucie: Non, pas de problème.\nPierre: Pff. Je ne comprends rien. Tu pourras m’expliquer après le cours ?\nLucie: Oui, si tu veux… On ira au café.\nPierre: Oui… euh non, j’ai oublié mon porte-monnaie \nLucie: Bon allez ! ce n’est pas grave, je t’invite.\nPierre: Tu es trop gentille.\nLucie: Oui, c’est bien possible."
metrics:
- rouge
model-index:
- name: Fine-tuned FLAN-T5 large model for French dialogue summarization
  results: []
---

<!-- This model card has been generated automatically according to the information the Trainer had access to. You
should probably proofread and complete it, then remove this comment. -->

# Fine-tuned FLAN-T5 large model for French dialogue summarization

This model is a fine-tuned version of [google/flan-t5-large](https://huggingface.co/google/flan-t5-large) for French dialogue summarization.

## Usage

Inference with 🤗 Pipeline

```python
import torch

from datasets import load_dataset
from transformers import pipeline

device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")

pipe = pipeline(
    "summarization",
    model="bofenghuang/flan-t5-large-french-dialogue-summarization",
    device=device,
)

dialogue_text = """
Pierre: J’ai oublié ma trousse. Tu peux me prêter un stylo.
Lucie: Tiens.
Pierre: Merci. Tu peux me donner une feuille de papier aussi ?
Lucie: Euh… oui. Tiens.
Pierre: Merci. Ça t’ennuie pas si je regarde avec toi ? J’ai oublié mon livre…
Lucie: Non, pas de problème.
Pierre: Pff. Je ne comprends rien. Tu pourras m’expliquer après le cours ?
Lucie: Oui, si tu veux… On ira au café.
Pierre: Oui… euh non, j’ai oublié mon porte-monnaie.
Lucie: Bon allez ! ce n’est pas grave, je t’invite.
Pierre: Tu es trop gentille.
Lucie: Oui, c’est bien possible.
"""

summarized_text = pipe(dialogue_text, max_length=1024)[0]["summary_text"]  # greedy
# summarized_text = pipe(dialogue_text, max_length=1024, num_beams=5)[0]["summary_text"]  # beam search
```