sharmax-vikas
commited on
Update README.md
Browse files
README.md
CHANGED
@@ -82,4 +82,31 @@ The following hyperparameters were used during training:
|
|
82 |
- Transformers 4.44.2
|
83 |
- Pytorch 2.4.0
|
84 |
- Datasets 3.0.0
|
85 |
-
- Tokenizers 0.19.1
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
82 |
- Transformers 4.44.2
|
83 |
- Pytorch 2.4.0
|
84 |
- Datasets 3.0.0
|
85 |
+
- Tokenizers 0.19.1
|
86 |
+
|
87 |
+
|
88 |
+
### How to use
|
89 |
+
```
|
90 |
+
from transformers import pipeline
|
91 |
+
|
92 |
+
pipe = pipeline("summarization", model="sharmax-vikas/flan-t5-base-samsum")
|
93 |
+
|
94 |
+
res = pipe('''dialogue:
|
95 |
+
Margaret: Hi, in December I'd like to meet on 4th and 11th around 10:00 or 11:00.
|
96 |
+
Evans: Hi, 4th - we can meet at 10:00.
|
97 |
+
Evans: And 11th - at 11:00.
|
98 |
+
Margaret: Okey. And what about 18th?
|
99 |
+
Evans: I'm not sure about 18th.
|
100 |
+
Evans: I might not be in town.
|
101 |
+
Margaret: Okey, so we'll see.
|
102 |
+
Evans: Yes. And I'll let you know next week.
|
103 |
+
Margaret: If it's not 18th, maybe we could meet on 17th?
|
104 |
+
Evans: If I go away, I won't also be 17th.
|
105 |
+
Margaret: Okey, I get it.
|
106 |
+
Evans: But we could meet 14th, if you like?
|
107 |
+
Margaret: Hm, I'm not sure whether I'm avaliable.
|
108 |
+
Evans: So let's set these dates later, ok?
|
109 |
+
Margaret: Okey and we see each other 4th 10:00.
|
110 |
+
Evans: Yes!''')
|
111 |
+
|
112 |
+
print(f"flan-t5-base summary:\n{res[0]['summary_text']}")```
|