shahules786
/

Safetybot-T5-base

Text Generation

text2text-generation

text-generation-inference

Inference Endpoints

Model card Files Files and versions Community

shahules786 commited on Mar 9, 2023

Commit

fb5db40

·

1 Parent(s): 80a8b88

Update README.md

Files changed (1) hide show

README.md +3 -3

README.md CHANGED Viewed

@@ -2,9 +2,9 @@
 A generative model trained to classify prompts into various safety categories and generate rules of thumb.
 ## Training
-Model architecture: `T5ForConditionalGeneration`
-Data : [prosocial-dialog](https://huggingface.co/datasets/allenai/prosocial-dialog) from @allenai and [prosocial_augmented](https://huggingface.co/datasets/shahules786/prosocial_augmented) from @shahules786
-Data preparation: model takes current user input and past conversations
 ## Example

 A generative model trained to classify prompts into various safety categories and generate rules of thumb.
 ## Training
+* Model architecture: `T5ForConditionalGeneration`
+* Data : [prosocial-dialog](https://huggingface.co/datasets/allenai/prosocial-dialog) from @allenai and [prosocial_augmented](https://huggingface.co/datasets/shahules786/prosocial_augmented) from @shahules786
+* Data preparation: model takes current user input and past conversations
 ## Example