Update model dropdown to show only Maya LoRA, Mistral v0.3, and Kimi K2

src/model_interface.py

@@ -43,58 +43,32 @@ class ModelInterface:
         
         logger.info(f"Model interface initialized on device: {self.device}")
         
-        # Define available models
+        # Define available models (optimized for HuggingFace Spaces)
         self.available_models = {
-            # Small/Fast models for quick testing
-            "microsoft/DialoGPT-small": {
-                "name": "DialoGPT Small",
-                "description": "Fast conversational model for testing",
-                "size": "Small (~300MB)",
-                "type": "local"
-            },
-            "microsoft/DialoGPT-medium": {
-                "name": "DialoGPT Medium", 
-                "description": "Balanced conversational model",
-                "size": "Medium (~1GB)",
-                "type": "local"
-            },
-            
-            # Larger models (will use quantization)
-            "meta-llama/Llama-2-7b-chat-hf": {
-                "name": "Llama 2 7B Chat",
-                "description": "Meta's Llama 2 optimized for chat",
-                "size": "Large (~7B params)",
-                "type": "local",
+            # Maya's fine-tuned LoRA model
+            "blakeurmos/maya-7b-lora-v1": {
+                "name": "Maya 7B (Fine-tuned)",
+                "description": "Maya's personality fine-tuned on Mistral-7B",
+                "size": "LoRA (~14MB + base model)",
+                "type": "lora",
+                "base_model": "mistralai/Mistral-7B-Instruct-v0.3",
                 "requires_auth": True
             },
-            "mistralai/Mistral-7B-Instruct-v0.1": {
-                "name": "Mistral 7B Instruct",
-                "description": "Mistral's instruction-tuned model",
+            # Latest Mistral instruction model
+            "mistralai/Mistral-7B-Instruct-v0.3": {
+                "name": "Mistral 7B Instruct v0.3",
+                "description": "Mistral's latest instruction-tuned model",
                 "size": "Large (~7B params)", 
-                "type": "local"
-            },
-            
-            # HF Inference API models (no local loading required)
-            "gpt2": {
-                "name": "GPT-2",
-                "description": "OpenAI's GPT-2 (via HF Inference API)",
-                "size": "API",
-                "type": "inference_api"
-            },
-            "microsoft/DialoGPT-large": {
-                "name": "DialoGPT Large",
-                "description": "Large conversational model (via API)",
-                "size": "API", 
-                "type": "inference_api"
+                "type": "inference_api",
+                "requires_auth": True
             },
-            
-            # Placeholder for fine-tuned models
-            "blakeurmos/maya-finetuned-v1": {
-                "name": "Maya Fine-tuned v1",
-                "description": "Custom fine-tuned Maya model",
-                "size": "Custom",
-                "type": "custom",
-                "exists": False  # Will check if exists
+            # Moonshot AI's latest model
+            "moonshotai/Kimi-K2-Instruct": {
+                "name": "Kimi K2 Instruct",
+                "description": "Moonshot AI's latest instruction model",
+                "size": "Large",
+                "type": "inference_api",
+                "requires_auth": True
             }
         }