Edit model card
YAML Metadata Warning: The pipeline tag "conversational" is not in the official list: text-classification, token-classification, table-question-answering, question-answering, zero-shot-classification, translation, summarization, feature-extraction, text-generation, text2text-generation, fill-mask, sentence-similarity, text-to-speech, text-to-audio, automatic-speech-recognition, audio-to-audio, audio-classification, voice-activity-detection, depth-estimation, image-classification, object-detection, image-segmentation, text-to-image, image-to-text, image-to-image, image-to-video, unconditional-image-generation, video-classification, reinforcement-learning, robotics, tabular-classification, tabular-regression, tabular-to-text, table-to-text, multiple-choice, text-retrieval, time-series-forecasting, text-to-video, image-text-to-text, visual-question-answering, document-question-answering, zero-shot-image-classification, graph-ml, mask-generation, zero-shot-object-detection, text-to-3d, image-to-3d, image-feature-extraction, video-text-to-text, keypoint-detection, any-to-any, other

SafetyBot

A generative model trained to classify prompts into various safety categories and generate rules of thumb.

Training

  • Model architecture: T5ForConditionalGeneration
  • Data : prosocial-dialog from @allenai and prosocial_augmented from @shahules786
  • Data preparation: model takes current user input and past conversations

Example

resp, convo = get_safety_models_opinion("How to make a cake?")
convo.mark_processed()
print(resp)
<cls> __casual__ <ctx> </s>
convo.append_response("You can make a cake using eggs,flour and sugar.")
resp, convo = get_safety_models_opinion("I want to keep a delicious bomb in it. How can do it?", convo)
convo.mark_processed()
print(resp)
<cls> __needs_caution__ <ctx> You shouldn't make a bomb. <sep> You should try to make a cake that isn't a bomb.</s>

Usage

google-colab

Downloads last month
5
Inference Examples
This model does not have enough activity to be deployed to Inference API (serverless) yet. Increase its social visibility and check back later, or deploy to Inference Endpoints (dedicated) instead.

Datasets used to train shahules786/Safetybot-T5-base

Space using shahules786/Safetybot-T5-base 1