Update gradio_app/config.py

gradio_app/config.py

@@ -1,44 +1,75 @@
-import logging
-
-# Configure logging
-logging.basicConfig(level=logging.INFO)
-logger = logging.getLogger(__name__)
-
-# LoRA configurations
-LORA_CONFIGS = {
-    "Gemma-3-1B-Instruct-Vi-Medical-LoRA": {
-        "base_model": "unsloth/gemma-3-1b-it",
-        "lora_adapter": "danhtran2mind/Gemma-3-1B-Instruct-Vi-Medical-LoRA"
-    },
-    "Gemma-3-1B-GRPO-Vi-Medical-LoRA": {
-        "base_model": "unsloth/gemma-3-1b-it",
-        "lora_adapter": "danhtran2mind/Gemma-3-1B-GRPO-Vi-Medical-LoRA"
-    },
-    "Llama-3.2-3B-Instruct-Vi-Medical-LoRA": {
-        "base_model": "unsloth/Llama-3.2-3B-Instruct",
-        "lora_adapter": "danhtran2mind/Llama-3.2-3B-Instruct-Vi-Medical-LoRA"
-    },
-    "Llama-3.2-1B-Instruct-Vi-Medical-LoRA": {
-        "base_model": "unsloth/Llama-3.2-1B-Instruct",
-        "lora_adapter": "danhtran2mind/Llama-3.2-1B-Instruct-Vi-Medical-LoRA"
-    },
-    "Llama-3.2-3B-Reasoning-Vi-Medical-LoRA": {
-        "base_model": "unsloth/Llama-3.2-3B-Instruct",
-        "lora_adapter": "danhtran2mind/Llama-3.2-3B-Reasoning-Vi-Medical-LoRA"
-    },
-    "Qwen-3-0.6B-Instruct-Vi-Medical-LoRA": {
-        "base_model": "Qwen/Qwen3-0.6B",
-        "lora_adapter": "danhtran2mind/Qwen-3-0.6B-Instruct-Vi-Medical-LoRA"
-    },
-    "Qwen-3-0.6B-Reasoning-Vi-Medical-LoRA": {
-        "base_model": "Qwen/Qwen3-0.6B",
-        "lora_adapter": "danhtran2mind/Qwen-3-0.6B-Reasoning-Vi-Medical-LoRA"
-    }
-}
-
-# Model settings
-MAX_INPUT_TOKEN_LENGTH = 4096
-DEFAULT_MAX_NEW_TOKENS = 512
-MAX_MAX_NEW_TOKENS = 2048
-
+import logging
+
+# Configure logging
+logging.basicConfig(level=logging.INFO)
+logger = logging.getLogger(__name__)
+
+# LoRA configurations
+# LORA_CONFIGS = {
+#     "Gemma-3-1B-Instruct-Vi-Medical-LoRA": {
+#         "base_model": "unsloth/gemma-3-1b-it",
+#         "lora_adapter": "danhtran2mind/Gemma-3-1B-Instruct-Vi-Medical-LoRA"
+#     },
+#     "Gemma-3-1B-GRPO-Vi-Medical-LoRA": {
+#         "base_model": "unsloth/gemma-3-1b-it",
+#         "lora_adapter": "danhtran2mind/Gemma-3-1B-GRPO-Vi-Medical-LoRA"
+#     },
+#     "Llama-3.2-3B-Instruct-Vi-Medical-LoRA": {
+#         "base_model": "unsloth/Llama-3.2-3B-Instruct",
+#         "lora_adapter": "danhtran2mind/Llama-3.2-3B-Instruct-Vi-Medical-LoRA"
+#     },
+#     "Llama-3.2-1B-Instruct-Vi-Medical-LoRA": {
+#         "base_model": "unsloth/Llama-3.2-1B-Instruct",
+#         "lora_adapter": "danhtran2mind/Llama-3.2-1B-Instruct-Vi-Medical-LoRA"
+#     },
+#     "Llama-3.2-3B-Reasoning-Vi-Medical-LoRA": {
+#         "base_model": "unsloth/Llama-3.2-3B-Instruct",
+#         "lora_adapter": "danhtran2mind/Llama-3.2-3B-Reasoning-Vi-Medical-LoRA"
+#     },
+#     "Qwen-3-0.6B-Instruct-Vi-Medical-LoRA": {
+#         "base_model": "Qwen/Qwen3-0.6B",
+#         "lora_adapter": "danhtran2mind/Qwen-3-0.6B-Instruct-Vi-Medical-LoRA"
+#     },
+#     "Qwen-3-0.6B-Reasoning-Vi-Medical-LoRA": {
+#         "base_model": "Qwen/Qwen3-0.6B",
+#         "lora_adapter": "danhtran2mind/Qwen-3-0.6B-Reasoning-Vi-Medical-LoRA"
+#     }
+# }
+
+LORA_CONFIGS = {
+    "Gemma-3-1B-Instruct-Vi-Medical-LoRA": {
+        "base_model": "google/gemma-3-1b-it",
+        "lora_adapter": "danhtran2mind/Gemma-3-1B-Instruct-Vi-Medical-LoRA"
+    },
+    "Gemma-3-1B-GRPO-Vi-Medical-LoRA": {
+        "base_model": "google/gemma-3-1b-it",
+        "lora_adapter": "danhtran2mind/Gemma-3-1B-GRPO-Vi-Medical-LoRA"
+    },
+    "Llama-3.2-3B-Instruct-Vi-Medical-LoRA": {
+        "base_model": "meta-llama/Llama-3.2-3B-Instruct",
+        "lora_adapter": "danhtran2mind/Llama-3.2-3B-Instruct-Vi-Medical-LoRA"
+    },
+    "Llama-3.2-1B-Instruct-Vi-Medical-LoRA": {
+        "base_model": "meta-llama/Llama-3.2-1B-Instruct",
+        "lora_adapter": "danhtran2mind/Llama-3.2-1B-Instruct-Vi-Medical-LoRA"
+    },
+    "Llama-3.2-3B-Reasoning-Vi-Medical-LoRA": {
+        "base_model": "meta-llama/Llama-3.2-3B-Instruct",
+        "lora_adapter": "danhtran2mind/Llama-3.2-3B-Reasoning-Vi-Medical-LoRA"
+    },
+    "Qwen-3-0.6B-Instruct-Vi-Medical-LoRA": {
+        "base_model": "Qwen/Qwen3-0.6B",
+        "lora_adapter": "danhtran2mind/Qwen-3-0.6B-Instruct-Vi-Medical-LoRA"
+    },
+    "Qwen-3-0.6B-Reasoning-Vi-Medical-LoRA": {
+        "base_model": "Qwen/Qwen3-0.6B",
+        "lora_adapter": "danhtran2mind/Qwen-3-0.6B-Reasoning-Vi-Medical-LoRA"
+    }
+}
+
+# Model settings
+MAX_INPUT_TOKEN_LENGTH = 4096
+DEFAULT_MAX_NEW_TOKENS = 512
+MAX_MAX_NEW_TOKENS = 2048
+
 MODEL_IDS = list(LORA_CONFIGS.keys())