Update app.py

app.py

@@ -14,18 +14,19 @@ login(token=HF_TOKEN)
 # Define models
 MODELS = {
     "athena-1": {
-        "name": "⚡ Atlas-Flash 1205",
+        "name": "🦁 Atlas-Flash 1215",
         "sizes": {
             "1.5B": "Spestly/Atlas-R1-1.5B-Preview",
         },
-        "emoji": "⚡",
+        "emoji": "",
         "experimental": True,
-        "is_vision": False,  # Enable vision support for this model
+        "is_vision": False,  # Set to True to enable vision support
+        "system_prompt_env": "ATLAS_FLASH_1215",  # Env variable for the system prompt
     },
 }
 
 # Profile pictures
-USER_PFP = "user.png"  # Hugging Face user avatar
+USER_PFP = "https://huggingface.co/front/assets/avatars.png"  # Hugging Face user avatar
 AI_PFP = "ai_pfp.png"  # Replace with the path to your AI's image or a URL
 
 class AtlasInferenceApp:
@@ -80,6 +81,7 @@ class AtlasInferenceApp:
                 "config": {
                     "name": f"{MODELS[model_key]['name']} {model_size}",
                     "path": model_path,
+                    "system_prompt": os.getenv(MODELS[model_key]["system_prompt_env"]),  # Load system prompt from env
                 }
             })
             return f"✅ {MODELS[model_key]['name']} {model_size} loaded successfully!"
@@ -91,9 +93,13 @@ class AtlasInferenceApp:
             return "⚠️ Please select and load a model first"
 
         try:
-            # Add a system instruction to guide the model's behavior
-            system_instruction = os.getenv("SYS_PROMPT")
-            prompt = f"{system_instruction}\n\n### Instruction:\n{message}\n\n### Response:"
+            # Get the system prompt for the current model
+            system_prompt = st.session_state.current_model["config"]["system_prompt"]
+            if not system_prompt:
+                return "⚠️ System prompt not found for the selected model."
+
+            # Add the system instruction to guide the model's behavior
+            prompt = f"{system_prompt}\n\n### Instruction:\n{message}\n\n### Response:"
 
             inputs = st.session_state.current_model["tokenizer"](
                 prompt,