Update ai_logic.py

ai_logic.py

@@ -25,6 +25,7 @@ if not HF_TOKEN:
 MODELS = {
     "Mistral 7B": "mistralai/Mistral-7B-Instruct-v0.3",
     "Phi-3 Mini (Microsoft)": "microsoft/Phi-3-mini-4k-instruct",
+    "Mistral (code) 7B": "mistralai/Mamba-Codestral-7B-v0.1",
     "Zephyr 7B": "HuggingFaceH4/zephyr-7b-beta"
 }