adding hate

app.py

@@ -10,6 +10,7 @@ SPAM_MODEL = "valurank/distilroberta-spam-comments-detection"
 TOXIC_MODEL = "s-nlp/roberta_toxicity_classifier"
 SENTIMENT_MODEL =  "nlptown/bert-base-multilingual-uncased-sentiment"
 NSFW_MODEL = "michellejieli/NSFW_text_classifier"
+HATE_MODEL = "facebook/roberta-hate-speech-dynabench-r4-target"
 
 spam = pipeline("text-classification", model=SPAM_MODEL)
 
@@ -19,6 +20,8 @@ sentiment = pipeline("text-classification", model = SENTIMENT_MODEL)
 
 nsfw = pipeline("text-classification", model = NSFW_MODEL)
 
+hate = pipeline("text-classification", model = HATE_MODEL)
+
 
 app = FastAPI()
 
@@ -49,6 +52,12 @@ def predict_nsfw(query: Query):
     result = nsfw(query.text)[0]
     return {"label": result["label"], "score": result["score"]}
 
+@app.post("/hate")
+def predict_hate(query: Query):
+    result = hate(query.text)[0]
+    return {"label": result["label"], "score": result["score"]}
+
+
 @app.get("/health")
 def health_check():