Update README.md
Browse files
README.md
CHANGED
@@ -53,7 +53,33 @@ as of April 8, 2025.
|
|
53 |
|
54 |
## Intended uses & limitations
|
55 |
|
56 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
57 |
|
58 |
## Training and evaluation data
|
59 |
|
|
|
53 |
|
54 |
## Intended uses & limitations
|
55 |
|
56 |
+
This model is designed for abstractive dialogue summarization. It can take in multi-turn conversations and generate concise summaries.
|
57 |
+
|
58 |
+
## How to Use
|
59 |
+
|
60 |
+
```Python
|
61 |
+
from transformers import pipeline
|
62 |
+
|
63 |
+
# Load summarization pipeline
|
64 |
+
model_name = "avanishd/pegasus-finetuned-samsum/"
|
65 |
+
summarizer = pipeline("summarization", model=model_name, tokenizer=model_name)
|
66 |
+
|
67 |
+
# Sample conversation
|
68 |
+
dialogue = """
|
69 |
+
John: Hey, are you free tomorrow?
|
70 |
+
Alice: I think so, why?
|
71 |
+
John: Want to catch a movie or grab lunch?
|
72 |
+
Alice: Sure, lunch sounds good. What time?
|
73 |
+
John: Let's say 1 PM at the new place downtown?
|
74 |
+
Alice: Works for me!
|
75 |
+
"""
|
76 |
+
|
77 |
+
# Generate summary
|
78 |
+
summary = summarizer(dialogue, max_length=60, min_length=15, do_sample=False)[0]['summary_text']
|
79 |
+
|
80 |
+
print("Summary:", summary)
|
81 |
+
|
82 |
+
```
|
83 |
|
84 |
## Training and evaluation data
|
85 |
|