gabrielaltay commited on
Commit
fe9941d
·
1 Parent(s): a81a191

remove nvidia nim

Browse files
Files changed (1) hide show
  1. app.py +3 -13
app.py CHANGED
@@ -58,28 +58,18 @@ CONGRESS_GOV_TYPE_MAP = {
58
  }
59
  OPENAI_CHAT_MODELS = [
60
  "gpt-4o-mini",
61
- "gpt-3.5-turbo-0125",
62
  "gpt-4o",
63
  ]
64
  ANTHROPIC_CHAT_MODELS = [
65
- "claude-3-opus-20240229",
66
- "claude-3-sonnet-20240229",
67
- "claude-3-5-sonnet-20240620",
68
  "claude-3-haiku-20240307",
 
 
69
  ]
70
- NVIDIA_NIM_CHAT_MODELS = [
71
- "microsoft/phi-3-mini-128k-instruct",
72
- "google/gemma-7b",
73
- "meta/llama3-8b-instruct",
74
- "meta/llama3-70b-instruct",
75
- "mistralai/mixtral-8x22b-instruct-v0.1",
76
- ]
77
- CHAT_MODELS = OPENAI_CHAT_MODELS + ANTHROPIC_CHAT_MODELS + NVIDIA_NIM_CHAT_MODELS
78
 
79
  PROVIDER_MODELS = {
80
  "OpenAI": OPENAI_CHAT_MODELS,
81
  "Anthropic": ANTHROPIC_CHAT_MODELS,
82
- "Nvidia NIM": NVIDIA_NIM_CHAT_MODELS,
83
  }
84
 
85
 
 
58
  }
59
  OPENAI_CHAT_MODELS = [
60
  "gpt-4o-mini",
 
61
  "gpt-4o",
62
  ]
63
  ANTHROPIC_CHAT_MODELS = [
 
 
 
64
  "claude-3-haiku-20240307",
65
+ "claude-3-5-sonnet-20240620",
66
+ "claude-3-opus-20240229",
67
  ]
68
+ CHAT_MODELS = OPENAI_CHAT_MODELS + ANTHROPIC_CHAT_MODELS
 
 
 
 
 
 
 
69
 
70
  PROVIDER_MODELS = {
71
  "OpenAI": OPENAI_CHAT_MODELS,
72
  "Anthropic": ANTHROPIC_CHAT_MODELS,
 
73
  }
74
 
75