Remove dropdown

app.py

@@ -23,6 +23,15 @@ login(token=HF_LE_LLM_READ_TOKEN)
 DEFAULT_MODEL = "le-llm/manipulative-score-model"
 DEVICE = "cuda"
 
+MODEL_OPTIONS = [
+    "le-llm/manipulative-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",
+
+]
 
 # --- Cache to avoid repeated reloads ---
 _model_cache: Dict[str, tuple[torch.nn.Module, AutoTokenizer]] = {}
@@ -63,29 +72,23 @@ def compute_score(text: str, model: torch.nn.Module, tokenizer: AutoTokenizer) -
 
 # --- Main scoring logic ---
 @spaces.GPU
-def bot(user_message: str, history: list[dict[str, Any]], model_choice: str):
+def bot(user_message: str, history: list[dict[str, Any]]):
     if not user_message.strip():
         return "", history
 
-    model, tokenizer = load_model(model_choice)  # returns embedding model
+    res = ""
     history = history + [{"role": "user", "content": user_message}]
-    score = compute_score(user_message, model, tokenizer)["score"]
+    for model_choice in MODEL_OPTIONS:
+        model, tokenizer = load_model(model_choice)  # returns embedding model
+        score = compute_score(user_message, model, tokenizer)["score"]
+        res += f"{model_choice}: {score}\n"
 
-    history.append({"role": "assistant", "content": f"{model_choice}: {score}"})
+    history.append({"role": "assistant", "content": res.strip()})
     return "", history
 
 # --- UI ---
 THEME = gr.themes.Soft(primary_hue="blue", secondary_hue="amber", neutral_hue="stone")
 
-MODEL_OPTIONS = [
-    "le-llm/manipulative-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():
     return "", []
@@ -94,14 +97,11 @@ def _clear_chat():
 with gr.Blocks(theme=THEME, fill_height=True) as demo:
     gr.Markdown("### 🤔 LAPA Quality Estimation")
 
-    with gr.Row():
-        model_choice = gr.Dropdown(MODEL_OPTIONS, value=DEFAULT_MODEL, label="Select Model")
-
     chatbot = gr.Chatbot(type="messages", height=480)
     msg = gr.Textbox(label=None, placeholder="Type your text…", lines=1)
     clear_btn = gr.Button("Clear")
 
-    msg.submit(bot, inputs=[msg, chatbot, model_choice], outputs=[msg, chatbot])
+    msg.submit(bot, inputs=[msg, chatbot], outputs=[msg, chatbot])
     clear_btn.click(_clear_chat, outputs=[msg, chatbot])
 
 if __name__ == "__main__":