add more models

app.py

@@ -51,11 +51,19 @@ MODELS_LIST = [
     # local models:
     ("====== IGCS Fine-tuned SLMs ======", None),
     ("Qwen2.5-3b-GenCS-union (local)", "shmuelamar/Qwen2.5-3b-GenCS-union"),
+    ("Qwen2.5-3b-GenCS-majority (local)", "shmuelamar/Qwen2.5-3b-GenCS-majority"),
+    ("Qwen2.5-7b-GenCS-union (local)", "shmuelamar/Qwen2.5-7b-GenCS-union"),
+    ("Qwen2.5-7b-GenCS-majority (local)", "shmuelamar/Qwen2.5-7b-GenCS-majority"),
+    ("Llama-3-8B-GenCS-union (local)", "shmuelamar/Llama-3-8B-GenCS-union"),
+    ("Llama-3-8B-GenCS-majority (local)", "shmuelamar/Llama-3-8B-GenCS-majority"),
+    ("SmolLM2-1.7B-GenCS-union (local)", "shmuelamar/SmolLM2-1.7B-GenCS-union"),
+    ("SmolLM2-1.7B-GenCS-majority (local)", "shmuelamar/SmolLM2-1.7B-GenCS-majority"),
     ("====== Zero-shot SLMs ======", None),
     ("Qwen/Qwen2.5-3B-Instruct (local)", "Qwen/Qwen2.5-3B-Instruct"),
     ("Qwen/Qwen2.5-7B-Instruct (local)", "Qwen/Qwen2.5-7B-Instruct"),
-    # API-based OpenRouter models:
-    ("====== API-based Models ======", None),
+    ("meta-llama/Meta-Llama-3-8B-Instruct (local)", "meta-llama/Meta-Llama-3-8B-Instruct"),
+    ("HuggingFaceTB/SmolLM2-1.7B-Instruct (local)", "HuggingFaceTB/SmolLM2-1.7B-Instruct"),
+    ("====== API-based Models (OpenRouter) ======", None),
     ("qwen/qwen3-14b (API)", "api:qwen/qwen3-14b:free"),
     ("moonshotai/kimi-k2 (API)", "api:moonshotai/kimi-k2:free"),
     ("deepseek/deepseek-chat-v3-0324 (API)", "api:deepseek/deepseek-chat-v3-0324:free"),
@@ -103,7 +111,7 @@ def completion(prompt: str, model_id: str):
     )
 
     # decode response
-    resp = tokenizer.decode(outputs[0][input_ids.shape[-1]:], skip_special_tokens=True)
+    resp = tokenizer.decode(outputs[0][input_ids.shape[-1] :], skip_special_tokens=True)
 
     # cleanup memory
     del model, tokenizer