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

Update README.md

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