Update inference.py

inference.py

@@ -1,21 +1,14 @@
-
 import joblib
-from sklearn.pipeline import Pipeline
-from sklearn.feature_extraction.text import TfidfVectorizer
-from sklearn.linear_model import LogisticRegression
-from typing import List
+from typing import List, Dict
 
 # Загрузка модели
 model_path = "model/language_classifier.joblib"
 model = joblib.load(model_path)
 
 # Функция для предсказания
-def predict(texts: List[str]):
+def predict(texts: List[str]) -> List[Dict[str, float]]:
     predictions = model.predict(texts)
-    return predictions
-
-def preprocess(text):
-    return [text]
-
-def postprocess(prediction):
-    return {"label": int(prediction[0])}
+    results = []
+    for prediction in predictions:
+        results.append({"label": str(prediction), "score": 1.0})
+    return results