health check

app.py

@@ -42,4 +42,34 @@ def predict_sentiment(query: Query):
 @app.post("/nsfw")
 def predict_nsfw(query: Query):
     result = nsfw(query.text)[0]
-    return {"label": result["label"], "score": result["score"]}
+    return {"label": result["label"], "score": result["score"]}
+
+@app.get("/health")
+def health_check():
+    status = {"server": "running"}
+
+    try:
+        spam("test")
+        status["spam"] = "running"
+    except Exception as e:
+        status["spam"] = f"error: {str(e)}"
+
+    try:
+        toxic("test")
+        status["toxic"] = "running"
+    except Exception as e:
+        status["toxic"] = f"error: {str(e)}"
+
+    try:
+        sentiment("test")
+        status["sentiment"] = "running"
+    except Exception as e:
+        status["sentiment"] = f"error: {str(e)}"
+
+    try:
+        nsfw("test")
+        status["nsfw"] = "running"
+    except Exception as e:
+        status["nsfw"] = f"error: {str(e)}"
+
+    return status

test_health.py

@@ -0,0 +1,7 @@
+import requests
+
+url = "https://cjell-Demo.hf.space"
+
+if __name__ == "__main__":
+    response = requests.get(f"{url}/health", timeout=10)
+    print(response.json())

test_sentiment.py

@@ -2,7 +2,7 @@ import requests
 
 url = "https://cjell-Demo.hf.space/sentiment"
 
-payload = {"text": "The people working outside are loud"}
+payload = {"text": "I am very upset with the new policy."}
 
 response = requests.post(url, json=payload)
 

test_toxic.py

@@ -2,7 +2,7 @@ import requests
 
 url = "https://cjell-Demo.hf.space/toxic"
 
-payload = {"text": "I do not agree with the policies being put in place by our city council."}
+payload = {"text": "I hate you!"}
 
 response = requests.post(url, json=payload)