inference.py

import joblib
from typing import List, Dict

# Загрузка модели
model_path = "model/language_classifier.joblib"
model = joblib.load(model_path)

# Функция для предсказания
def predict(texts: List[str]) -> List[Dict[str, float]]:
    predictions = model.predict(texts)
    results = []
    for prediction in predictions:
        results.append({"label": str(prediction), "score": 1.0})
    return results