lxyuan commited on
Commit
6ef3fdb
1 Parent(s): 2fc49fd

Fix inference example code

Browse files
Files changed (1) hide show
  1. README.md +3 -1
README.md CHANGED
@@ -61,7 +61,7 @@ raw_datasets["test"] = summary_test
61
  ```python
62
  from transformers import pipeline
63
 
64
- pipe = pipeline("text2text-generation", model="sshleifer/distilbart-cnn-12-6")
65
 
66
  text = """The tower is 324 metres (1,063 ft) tall, about the same height as
67
  an 81-storey building, and the tallest structure in Paris. Its base is square,
@@ -75,6 +75,8 @@ of the tower in 1957, it is now taller than the Chrysler Building by 5.2 metres
75
  free-standing structure in France after the Millau Viaduct.
76
  """
77
 
 
 
78
  >>>"""The Eiffel Tower is the tallest man-made structure in the world .
79
  The tower is 324 metres tall, about the same height as an 81-storey building .
80
  Due to the addition of a broadcasting aerial in 1957, it is now taller than
 
61
  ```python
62
  from transformers import pipeline
63
 
64
+ pipe = pipeline("text2text-generation", model="lxyuan/distilbart-finetuned-summarization")
65
 
66
  text = """The tower is 324 metres (1,063 ft) tall, about the same height as
67
  an 81-storey building, and the tallest structure in Paris. Its base is square,
 
75
  free-standing structure in France after the Millau Viaduct.
76
  """
77
 
78
+ pipe(text)
79
+
80
  >>>"""The Eiffel Tower is the tallest man-made structure in the world .
81
  The tower is 324 metres tall, about the same height as an 81-storey building .
82
  Due to the addition of a broadcasting aerial in 1957, it is now taller than