more models

.gitignore

@@ -0,0 +1,2 @@
+/test.py
+/test2.py

app.py

@@ -6,7 +6,13 @@ import os
 
 os.environ["HF_HOME"] = "/tmp" 
 
-classifier = pipeline("text-classification", model="mrm8488/bert-tiny-finetuned-sms-spam-detection")
+spam = pipeline("text-classification", model="mrm8488/bert-tiny-finetuned-sms-spam-detection")
+
+toxic = pipeline("text-classification", model="cardiffnlp/twitter-roberta-base-toxic")
+
+sentiment = pipeline("text-classification", model = "nlptown/bert-base-multilingual-uncased-sentiment")
+
+
 
 app = FastAPI()
 
@@ -17,8 +23,17 @@ def root():
 class Query(BaseModel):
     text: str
 
-@app.post("/predict")
-def predict(query: Query):
-    result = classifier(query.text)[0]
+@app.post("/spam")
+def predict_spam(query: Query):
+    result = spam(query.text)[0]
+    return {"label": result["label"], "score": result["score"]}
+
+@app.post("/toxic")
+def predict_toxic(query: Query):
+    result = toxic(query.text)[0]
     return {"label": result["label"], "score": result["score"]}
 
+@app.post("/sentiment")
+def predict_sentiment(query: Query):
+    result = toxic(query.text)[0]
+    return {"label": result["label"], "score": result["score"]}

test.py → test_spam.py

@@ -1,6 +1,6 @@
 import requests
 
-url = "https://cjell-Demo.hf.space/predict"
+url = "https://cjell-Demo.hf.space/spam"
 
 payload = {"text": "Congragulations you have won $1000"}
 

test_toxic.py

@@ -0,0 +1,13 @@
+import requests
+
+url = "https://cjell-Demo.hf.space/predict"
+
+payload = {"text": "I Hate this I want to do bad things you suck!"}
+
+response = requests.post(url, json=payload)
+
+print("Status:", response.status_code)
+try:
+    print("JSON:", response.json())
+except Exception:
+    print("Raw text:", response.text)