Update app.py

app.py

@@ -1,4 +1,4 @@
-# Multi-agent UI generator with lazy model loading to reduce memory usage
+# Multi-agent UI generator with correct base models for each agent
 
 import streamlit as st
 import time
@@ -16,23 +16,23 @@ HF_TOKEN = os.getenv("HF_TOKEN")
 
 AGENT_MODEL_CONFIG = {
     "product_manager": {
-        "base": "google/gemma-1.1-7b-it",
-        "adapter": "spandana30/project-manager-gemma"
+        "base": "unsloth/mistral-7b-bnb-4bit",
+        "adapter": "spandana30/product-manager-mistral"
     },
     "project_manager": {
-        "base": "google/gemma-1.1-7b-it",
+        "base": "unsloth/gemma-3-1b-it",
         "adapter": "spandana30/project-manager-gemma"
     },
     "software_architect": {
-        "base": "cohere/command-r",
+        "base": "unsloth/c4ai-command-r-08-2024-bnb-4bit",
         "adapter": "spandana30/software-architect-cohere"
     },
     "software_engineer": {
-        "base": "codellama/CodeLLaMA-7b-Instruct-hf",
+        "base": "codellama/CodeLLaMA-7b-hf",
         "adapter": "spandana30/software-engineer-codellama"
     },
     "qa": {
-        "base": "codellama/CodeLLaMA-7b-Instruct-hf",
+        "base": "codellama/CodeLLaMA-7b-hf",
         "adapter": "spandana30/software-engineer-codellama"
     },
 }
@@ -180,4 +180,4 @@ def main():
                 st.write(f"🧩 {stage.replace('_', ' ').title()} Time: {final_state['timings'].get(stage, 0):.2f}s")
 
 if __name__ == "__main__":
-    main()
+    main()