Update scripts/classifier.py

scripts/classifier.py

@@ -1,4 +1,28 @@
+import tensorflow as tf
+from pathlib import Path
+from huggingface_hub import hf_hub_download
+import os
+
+# Если модель приватная — добавь HF_TOKEN в переменные окружения (Settings → Secrets)
+model_path = hf_hub_download(
+    repo_id="aikenml/AI-text-humanizer-deberta",
+    filename="deberta_classifier.keras",
+    # token=os.getenv("HF_TOKEN")  # можно убрать, если репозиторий публичный
+)
+
+# Загружаем модель (без компиляции, чтобы убрать предупреждения)
+model = tf.keras.models.load_model(model_path, compile=False)
 
 def debert_classifier(text):
     # ....
-    return 0.7
+    # Predict
+    predictions = model.predict([text])
+
+    # Show results
+    for text, pred in zip([text], predictions):
+        # label = "AI-generated" if pred > 0.5 else "Human-written"
+        # print(f"{text} --> {label} (score={pred[0]:.4f})")
+        # return label
+        # return "hello"
+        # return 0.7
+        return pred