Update README.md

README.md

@@ -15,11 +15,24 @@ tags:
 - toxicity-reduction
 
 widget:
-- text: "Summarize the following Conversation: Kate: Good morning. Kai: Hi! How official! Kate: I wrote it at 4am Kai: I've noticed. Why? Kate: I had to get up early to catch the bus to the airport Kai: Where are you flying? Kate: To Antwerp! I'm fed up with Cambridge Kai: poor thing. Why? Kate: Just a stupid, elitist place without a soul. Or with a soul made of money. Kai: Try to rest a bit in Belgium, do not work too much. Kate: I have to work, but at least not in this soulless place. Kai: When are you coming back? Kate: I have to see my supervisor on Monday <unk> Kai: not too long a break Kate: Still better than nothing. Summary:"
+- text: >-
+    Summarize the following Conversation: Kate: Good morning. Kai: Hi! How
+    official! Kate: I wrote it at 4am Kai: I've noticed. Why? Kate: I had to get
+    up early to catch the bus to the airport Kai: Where are you flying? Kate: To
+    Antwerp! I'm fed up with Cambridge Kai: poor thing. Why? Kate: Just a
+    stupid, elitist place without a soul. Or with a soul made of money. Kai: Try
+    to rest a bit in Belgium, do not work too much. Kate: I have to work, but at
+    least not in this soulless place. Kai: When are you coming back? Kate: I
+    have to see my supervisor on Monday <unk> Kai: not too long a break Kate:
+    Still better than nothing. Summary:
   example_title: Summarization Example 1
-- text: "Summarize the following Conversation: Dean: I feel sick Scott: hungover? Dean: no, like I ate something bad Scott: what did you eat yesterday? Dean: breakfast at Coffee Lovers' Scott: this is a rather safe place Dean: and Chinese from TaoTao for dinner Scott: now we have a suspect Summary:"
+- text: >-
+    Summarize the following Conversation: Dean: I feel sick Scott: hungover?
+    Dean: no, like I ate something bad Scott: what did you eat yesterday? Dean:
+    breakfast at Coffee Lovers' Scott: this is a rather safe place Dean: and
+    Chinese from TaoTao for dinner Scott: now we have a suspect Summary:
   example_title: Summarization Example 2
-
+  
 inference:
   parameters:
     max_new_tokens: 256
@@ -30,8 +43,9 @@ inference:
     num_beams: 3
     no_repeat_ngram_size: 2
     num_return_sequences: 1
-    do_sample: True
-
+    do_sample: true
+    
+pipeline_tag: summarization
 ---
 
 # Flan-T5 (base-sized) Dialogue Summarization with reduced toxicity using RLAIF
@@ -94,4 +108,4 @@ output = tokenizer.batch_decode(summary, skip_special_tokens=True)
 # "Dean ate breakfast at Coffee Lovers' yesterday and Chinese from TaoTao for dinner."
 ```
 
-> Designed and Developed with <span style="color: #e25555;">&hearts;</span> by [Praneet](https://deathreaper0965.github.io/) | [LinkedIn](http://linkedin.com/in/deathreaper0965) | [GitHub](https://github.com/DeathReaper0965/)
+> Designed and Developed with <span style="color: #e25555;">&hearts;</span> by [Praneet](https://deathreaper0965.github.io/) | [LinkedIn](http://linkedin.com/in/deathreaper0965) | [GitHub](https://github.com/DeathReaper0965/)