swapping spam

app.py

@@ -1,17 +1,17 @@
 from fastapi import FastAPI
 from fastapi.middleware.cors import CORSMiddleware
 from pydantic import BaseModel
-from transformers import pipeline
+from transformers.pipelines import pipeline
 import os
 
-
 os.environ["HF_HOME"] = "/tmp" 
 
-SPAM_MODEL = "valurank/distilroberta-spam-comments-detection"
+SPAM_MODEL = "cjell/spam-detector-roberta "
 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"
+IMAGE_MODEL = "Falconsai/nsfw_image_detection"
 
 spam = pipeline("text-classification", model=SPAM_MODEL)
 
@@ -23,6 +23,8 @@ nsfw = pipeline("text-classification", model = NSFW_MODEL)
 
 hate = pipeline("text-classification", model = HATE_MODEL)
 
+image = pipeline("image-classification", model = IMAGE_MODEL)
+
 
 app = FastAPI()