bofenghuang
commited on
Commit
·
2d54efe
1
Parent(s):
dc2931f
Update README.md
Browse files
README.md
CHANGED
@@ -17,7 +17,7 @@ model-index:
|
|
17 |
<!-- This model card has been generated automatically according to the information the Trainer had access to. You
|
18 |
should probably proofread and complete it, then remove this comment. -->
|
19 |
|
20 |
-
# Fine-tuned FLAN-T5 large model for French
|
21 |
|
22 |
This model is a fine-tuned version of [google/flan-t5-large](https://huggingface.co/google/flan-t5-large) for French dialogue summarization.
|
23 |
|
@@ -27,15 +27,13 @@ Inference with 🤗 Pipeline
|
|
27 |
|
28 |
```python
|
29 |
import torch
|
30 |
-
|
31 |
-
from datasets import load_dataset
|
32 |
from transformers import pipeline
|
33 |
|
34 |
device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
|
35 |
|
36 |
pipe = pipeline(
|
37 |
"summarization",
|
38 |
-
model="bofenghuang/flan-t5-large-
|
39 |
device=device,
|
40 |
)
|
41 |
|
|
|
17 |
<!-- This model card has been generated automatically according to the information the Trainer had access to. You
|
18 |
should probably proofread and complete it, then remove this comment. -->
|
19 |
|
20 |
+
# Fine-tuned FLAN-T5 large model for French Dialogue Summarization
|
21 |
|
22 |
This model is a fine-tuned version of [google/flan-t5-large](https://huggingface.co/google/flan-t5-large) for French dialogue summarization.
|
23 |
|
|
|
27 |
|
28 |
```python
|
29 |
import torch
|
|
|
|
|
30 |
from transformers import pipeline
|
31 |
|
32 |
device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
|
33 |
|
34 |
pipe = pipeline(
|
35 |
"summarization",
|
36 |
+
model="bofenghuang/flan-t5-large-dialogsum-fr",
|
37 |
device=device,
|
38 |
)
|
39 |
|