Spaces:
Running
Running
File size: 957 Bytes
d68e65a |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 |
from pydantic import BaseModel
from typing import List
from app.models.check_type import CheckType
from app.services.text_cleaner import clean_text
from app.ml_models.classifier import Classifier
from app.ml_models.gemini_moderator import gemini_for_cleansing
from app.ml_models.gemini_moderator import gemini_for_misinfo_detection
classifier = Classifier()
classifier.initialize_classifier()
class ModerationRequest(BaseModel):
content: str
checkFor: List[CheckType]
threshold: float = 0.6
def correct_typos(self) -> None:
self.content = clean_text(self.content)
def classify_moderation(self) -> None:
result = classifier.predict_nsfw(self.content)
return result
def cleanse_content(self) -> str:
return gemini_for_cleansing.cleanse(self.content)
def identify_misinfo(self) -> str:
return gemini_for_misinfo_detection.check_misinfo(self.content)
|