azettl commited on
Commit
46fb991
Β·
1 Parent(s): 16c3e93
Files changed (1) hide show
  1. app.py +8 -15
app.py CHANGED
@@ -93,30 +93,27 @@ class VisualConsensusEngine:
93
  self.moderator_model = moderator_model or MODERATOR_MODEL
94
  self.search_agent = WebSearchAgent()
95
  self.update_callback = update_callback # For real-time updates
96
-
97
- mistral_key = os.getenv("MISTRAL_API_KEY")
98
- sambanova_key = os.getenv("SAMBANOVA_API_KEY")
99
 
100
  self.models = {
101
  'mistral': {
102
  'name': 'Mistral Large',
103
- 'api_key': mistral_key,
104
- 'available': bool(mistral_key)
105
  },
106
  'sambanova_deepseek': {
107
  'name': 'DeepSeek-R1',
108
- 'api_key': sambanova_key,
109
- 'available': bool(sambanova_key)
110
  },
111
  'sambanova_llama': {
112
  'name': 'Meta-Llama-3.1-8B',
113
- 'api_key': sambanova_key,
114
- 'available': bool(sambanova_key)
115
  },
116
  'sambanova_qwq': {
117
  'name': 'QwQ-32B',
118
- 'api_key': sambanova_key,
119
- 'available': bool(sambanova_key)
120
  },
121
  'search': {
122
  'name': 'Web Search Agent',
@@ -855,10 +852,6 @@ with gr.Blocks(title="🎭 Consilium: Visual AI Consensus Platform", theme=gr.th
855
  with gr.Tab("πŸ”§ Configuration & Setup"):
856
  def check_model_status():
857
  engine = VisualConsensusEngine()
858
-
859
- mistral_key = os.getenv("MISTRAL_API_KEY")
860
- sambanova_key = os.getenv("SAMBANOVA_API_KEY")
861
-
862
  status_info = "## πŸ” Model Availability Status\n\n"
863
 
864
  for model_id, model_info in engine.models.items():
 
93
  self.moderator_model = moderator_model or MODERATOR_MODEL
94
  self.search_agent = WebSearchAgent()
95
  self.update_callback = update_callback # For real-time updates
 
 
 
96
 
97
  self.models = {
98
  'mistral': {
99
  'name': 'Mistral Large',
100
+ 'api_key': MISTRAL_API_KEY,
101
+ 'available': bool(MISTRAL_API_KEY)
102
  },
103
  'sambanova_deepseek': {
104
  'name': 'DeepSeek-R1',
105
+ 'api_key': SAMBANOVA_API_KEY,
106
+ 'available': bool(SAMBANOVA_API_KEY)
107
  },
108
  'sambanova_llama': {
109
  'name': 'Meta-Llama-3.1-8B',
110
+ 'api_key': SAMBANOVA_API_KEY,
111
+ 'available': bool(SAMBANOVA_API_KEY)
112
  },
113
  'sambanova_qwq': {
114
  'name': 'QwQ-32B',
115
+ 'api_key': SAMBANOVA_API_KEY,
116
+ 'available': bool(SAMBANOVA_API_KEY)
117
  },
118
  'search': {
119
  'name': 'Web Search Agent',
 
852
  with gr.Tab("πŸ”§ Configuration & Setup"):
853
  def check_model_status():
854
  engine = VisualConsensusEngine()
 
 
 
 
855
  status_info = "## πŸ” Model Availability Status\n\n"
856
 
857
  for model_id, model_info in engine.models.items():