better health check

app.py

@@ -46,30 +46,30 @@ def predict_nsfw(query: Query):
 
 @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)}"
+
+    status = {
+        "server": "running",
+        "models": {}
+    }
+
+    models = {
+        "spam": ("valurank/distilroberta-spam-comments-detection", spam),
+        "toxic": ("s-nlp/roberta_toxicity_classifier", toxic),
+        "sentiment": ("nlptown/bert-base-multilingual-uncased-sentiment", sentiment),
+        "nsfw": ("michellejieli/NSFW_text_classifier", nsfw),
+    }
+
+    for key, (model_name, model_pipeline) in models.items():
+        try:
+            model_pipeline("test")
+            status["models"][key] = {
+                "model_name": model_name,
+                "status": "running"
+            }
+        except Exception as e:
+            status["models"][key] = {
+                "model_name": model_name,
+                "status": f"error: {str(e)}"
+            }
 
     return status