Update README.md

README.md

@@ -35,17 +35,26 @@ pip install llm-toys
 ```
 
 ```python
-from llm_toys.tasks import Paraphraser
-
-paraphraser = Paraphraser()
-paraphraser.paraphrase("Hey, can yuo hepl me cancel my last order?")
-# "Could you kindly assist me in canceling my previous order?"
-
-paraphraser.paraphrase("Hey, can yuo hepl me cancel my last order?", tone="professional")
-# "I would appreciate guidance on canceling my previous order."
-
-paraphraser.paraphrase("Hey, can yuo hepl me cancel my last order?", tone="witty")
-# "Hey, I need your help with my last order. Can you wave your magic wand and make it disappear?"
+from llm_toys.tasks import SummaryAndTopicGenerator
+
+summary_theme_generator = SummaryAndTopicGenerator()
+summary_theme_generator.generate_summary_and_topic(
+    """
+    #Person1#: I'm so excited for the premiere of the latest Studio Ghibli movie!
+    #Person2#: What's got you so hyped?
+    #Person1#: Studio Ghibli movies are pure magic! The animation, storytelling, everything is incredible.
+    #Person2#: Which movie is it?
+    #Person1#: It's called "Whisper of the Wind." It's about a girl on a magical journey to save her village.
+    #Person2#: Sounds amazing! I'm in for the premiere.
+    #Person1#: Great! We're in for a visual masterpiece and a heartfelt story.
+    #Person2#: Can't wait to be transported to their world.
+    #Person1#: It'll be an unforgettable experience, for sure!
+    """.strip()
+)
+# {"summary": "#Person1# is excited for the premiere of the latest Studio Ghibli movie.
+#              #Person1# thinks the animation, storytelling, and heartfelt story will be unforgettable.
+#              #Person2# is also excited for the premiere.",
+#  "topic": "Studio ghibli movie"}
 ```
 
 ## Training params