Add more score models

app.py

@@ -65,7 +65,7 @@ def compute_score(text: str, model: torch.nn.Module, tokenizer: AutoTokenizer) -
             logits = outputs.logits.squeeze(-1).float().cpu().numpy()
 
         res = {}
-        res["score"] = logits.tolist()
+        res["score"] = logits.tolist()[0]
         res["int_score"] = [int(round(max(0, min(score, 5)))) for score in logits]
         return res
 
@@ -87,7 +87,12 @@ THEME = gr.themes.Soft(primary_hue="blue", secondary_hue="amber", neutral_hue="s
 
 MODEL_OPTIONS = [
     "le-llm/manipulative-score-model",
-    "le-llm/gec-score-model"
+    "le-llm/gec-score-model",
+    "le-llm/fineweb-mixtral-edu-score",
+    "le-llm/fineweb-nemotron-edu-score",
+    "le-llm/alignment-score-model",
+    "le-llm/fasttext-quality-score",
+
 ]
 
 def _clear_chat():