File size: 302 Bytes
3ece550
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
from models.base import BaseModel

# Exemple de modèle minimal
class DummyModel(BaseModel):
    def predict(self, text):
        if "stupide" in text.lower():
            return "toxique"
        return "non-toxique"

dummy = DummyModel()

def predict(text: str) -> str:
    return dummy.predict(text)