Update app.py

app.py

@@ -31,6 +31,8 @@ MODEL_IDS: List[str] = [
     "CodeSoft/MetaDiffusion-150M-ChatBase",
     "BananaMind/BananaMind-2-Medium-Chat",
     "SupraLabs/Supra2-100M-Instruct",
+    "SupraLabs/Supra2-Medium-Instruct",
+    "SupraLabs/Supra1.5-50M-Instruct-exp",
     "HuggingFaceTB/SmolLM2-135M-Instruct",
     "OpenCerebral/Boris-1.3-125M-Instruct",
     "OpenCerebral/Boris-1.3-75M-Instruct",
@@ -43,6 +45,8 @@ MODEL_DISPLAY: Dict[str, str] = {
     "CodeSoft/MetaDiffusion-150M-ChatBase": "MetaDiffusion-150M-ChatBase",
     "BananaMind/BananaMind-2-Medium-Chat": "BananaMind-2-Medium-Chat",
     "SupraLabs/Supra2-100M-Instruct": "Supra2-100M-Instruct",
+    "SupraLabs/Supra2-Medium-Instruct": "Supra2-Medium-Instruct",
+    "SupraLabs/Supra1.5-50M-Instruct-exp": "Supra1.5-50M-Instruct-exp",
     "HuggingFaceTB/SmolLM2-135M-Instruct": "SmolLM2-135M-Instruct",
     "OpenCerebral/Boris-1.3-125M-Instruct": "Boris-1.3-125M-Instruct",
     "OpenCerebral/Boris-1.3-75M-Instruct": "Boris-1.3-75M-Instruct",
@@ -87,6 +91,8 @@ MODEL_PARAMS: Dict[str, float] = {
     "CodeSoft/MetaDiffusion-150M-ChatBase": 169.5e6,
     "BananaMind/BananaMind-2-Medium-Chat": 49.6e6,
     "SupraLabs/Supra2-100M-Instruct": 100.0e6,
+    "SupraLabs/Supra2-Medium-Instruct": 25.4e6,
+    "SupraLabs/Supra1.5-50M-Instruct-exp": 51.8e6,
     "HuggingFaceTB/SmolLM2-135M-Instruct": 135.0e6,
     "OpenCerebral/Boris-1.3-125M-Instruct": 125.0e6,
     "OpenCerebral/Boris-1.3-75M-Instruct": 75.0e6,
@@ -178,6 +184,8 @@ CHAT_FILE = get_chat_file()
 GEN_DEFAULTS: Dict[str, dict] = {
     "HuggingFaceTB/SmolLM2-135M-Instruct": {"max_new_tokens": 64, "temperature": 0.7, "top_p": 0.9, "repetition_penalty": 1.1, "do_sample": True},
     "SupraLabs/Supra2-100M-Instruct": {"max_new_tokens": 64, "temperature": 0.7, "top_p": 0.9, "top_k": 25, "repetition_penalty": 1.1, "do_sample": True, "no_repeat_ngram_size": 3},
+    "SupraLabs/Supra2-Medium-Instruct": {"max_new_tokens": 64, "temperature": 0.2, "top_p": 0.85, "top_k": 25, "do_sample": True, "no_repeat_ngram_size": 3},
+    "SupraLabs/Supra1.5-50M-Instruct-exp": {"max_new_tokens": 64, "temperature": 0.7, "top_p": 0.9, "repetition_penalty": 1.2, "do_sample": True},
     "BananaMind/BananaMind-2-Medium-Chat": {"max_new_tokens": 64, "temperature": 0.7, "top_p": 0.9, "repetition_penalty": 1.1, "do_sample": True},
     "CodeSoft/MetaDiffusion-150M-ChatBase": {"max_new_tokens": 96, "num_steps": 128, "temperature": 0.7, "top_p": 0.9, "repetition_penalty": 1.5},
     "fromziro/Zero-v0.1-150M": {"max_new_tokens": 64, "temperature": 0.8, "top_p": 0.95, "repetition_penalty": 1.1, "do_sample": True},
@@ -203,6 +211,8 @@ GEN_DEFAULTS: Dict[str, dict] = {
 MODEL_CONTEXT: Dict[str, int] = {
     "HuggingFaceTB/SmolLM2-135M-Instruct": 2048,
     "SupraLabs/Supra2-100M-Instruct": 1024,
+    "SupraLabs/Supra2-Medium-Instruct": 1024,
+    "SupraLabs/Supra1.5-50M-Instruct-exp": 5120,
     "BananaMind/BananaMind-2-Medium-Chat": 3072,
     "CodeSoft/MetaDiffusion-150M-ChatBase": 5120,
     "fromziro/Zero-v0.1-150M": 2048,
@@ -1383,4 +1393,4 @@ if __name__ == "__main__":
         logger.warning(f"Leaderboard preview failed: {e}")
     demo = create_demo()
     demo.queue(max_size=20)
-    demo.launch(server_name="0.0.0.0", server_port=7860, show_error=True, theme=gr.themes.Base(), css=CSS)
+    demo.launch(server_name="0.0.0.0", server_port=7860, show_error=True, theme=gr.themes.Base(), css=CSS)