nsfw and spam fix

app.py

@@ -6,12 +6,13 @@ import os
 
 os.environ["HF_HOME"] = "/tmp" 
 
-spam = pipeline("text-classification", model="Goodmotion/spam-mail-classifier")
+spam = pipeline("text-classification", model="valurank/distilroberta-spam-comments-detection")
 
 toxic = pipeline("text-classification", model="s-nlp/roberta_toxicity_classifier")
 
 sentiment = pipeline("text-classification", model = "nlptown/bert-base-multilingual-uncased-sentiment")
 
+nsfw = pipeline("text-classification", model = "michellejieli/NSFW_text_classifier")
 
 
 app = FastAPI()
@@ -36,4 +37,9 @@ def predict_toxic(query: Query):
 @app.post("/sentiment")
 def predict_sentiment(query: Query):
     result = sentiment(query.text)[0]
+    return {"label": result["label"], "score": result["score"]}
+
+@app.post("/nsfw")
+def predict_nsfw(query: Query):
+    result = nsfw(query.text)[0]
     return {"label": result["label"], "score": result["score"]}

test.nsfw.py

@@ -0,0 +1,13 @@
+import requests
+
+url = "https://cjell-Demo.hf.space/nsfw"
+
+payload = {"text": "Hello"}
+
+response = requests.post(url, json=payload)
+
+print("Status:", response.status_code)
+try:
+    print("JSON:", response.json())
+except Exception:
+    print("Raw text:", response.text)

test_spam.py

@@ -2,7 +2,7 @@ import requests
 
 url = "https://cjell-Demo.hf.space/spam"
 
-payload = {"text": "Congratulations! You've won a free iPhone 14. Click here to claim: http://bit.ly/free-iphone"}
+payload = {"text": "We are having a neighborhood clean up tomorrow if you want to join us."}
 
 response = requests.post(url, json=payload)