Update app.py

app.py

@@ -1,4 +1,4 @@
-# Updated multi-agent UI generation system with custom fine-tuned LoRA adapters
+# Updated multi-agent UI generation system with all agents using public base models (Gemma used for Product Manager too)
 
 import streamlit as st
 import time
@@ -14,8 +14,8 @@ st.set_page_config(page_title="Multi-Agent Collaboration", layout="wide")
 # Agent model loading config
 AGENT_MODEL_CONFIG = {
     "product_manager": {
-        "base": "mistralai/Mistral-7B-Instruct-v0.2",
-        "adapter": "spandana30/product-manager-mistral"
+        "base": "google/gemma-1.1-7b-it",
+        "adapter": "spandana30/project-manager-gemma"
     },
     "project_manager": {
         "base": "google/gemma-1.1-7b-it",
@@ -177,4 +177,4 @@ def main():
                 st.write(f"🧩 {stage.replace('_', ' ').title()} Time: {final_state['timings'].get(stage, 0):.2f}s")
 
 if __name__ == "__main__":
-    main()
+    main()