spandana30 commited on
Commit
ab6fadd
·
verified ·
1 Parent(s): 2b46ce7

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -1,4 +1,4 @@
1
- # Multi-agent UI generator with correct base models for each agent
2
 
3
  import streamlit as st
4
  import time
@@ -16,8 +16,8 @@ HF_TOKEN = os.getenv("HF_TOKEN")
16
 
17
  AGENT_MODEL_CONFIG = {
18
  "product_manager": {
19
- "base": "unsloth/mistral-7b-bnb-4bit",
20
- "adapter": "spandana30/product-manager-mistral"
21
  },
22
  "project_manager": {
23
  "base": "unsloth/gemma-3-1b-it",
 
1
+ # Multi-agent UI generator with Gemma model for both product and project manager agents
2
 
3
  import streamlit as st
4
  import time
 
16
 
17
  AGENT_MODEL_CONFIG = {
18
  "product_manager": {
19
+ "base": "unsloth/gemma-3-1b-it",
20
+ "adapter": "spandana30/project-manager-gemma"
21
  },
22
  "project_manager": {
23
  "base": "unsloth/gemma-3-1b-it",