matt HOFFNER commited on
Commit
00a6446
·
1 Parent(s): a21e1b7

more specific model names

Browse files
components/Playground/index.tsx CHANGED
@@ -154,8 +154,8 @@ const Playground = () => {
154
  onChange={(e) => setAIProvider(e.target.value)}
155
  className="border p-2 rounded-md shadow-sm w-full bg-transparent text-gray-700 mt-2"
156
  >
157
- <option value="openai">OpenAI</option>
158
- <option value="huggingface">Hugging Face</option>
159
  </select>
160
  </div>
161
  )}
 
154
  onChange={(e) => setAIProvider(e.target.value)}
155
  className="border p-2 rounded-md shadow-sm w-full bg-transparent text-gray-700 mt-2"
156
  >
157
+ <option value="openai">gpt-4</option>
158
+ <option value="huggingface">meta-llama/Llama-2-70b-chat-hf</option>
159
  </select>
160
  </div>
161
  )}
pages/api/chat/index.ts CHANGED
@@ -38,7 +38,7 @@ export default async function(req: Request) {
38
  } else if (aiProvider === 'huggingface') {
39
  const response = Hf.textGenerationStream({
40
  // @ts-ignore
41
- model: 'meta-llama/Llama-2-7b-chat-hf',
42
  inputs: experimental_buildLlama2Prompt(messages),
43
  parameters: {
44
  max_new_tokens: 500,
 
38
  } else if (aiProvider === 'huggingface') {
39
  const response = Hf.textGenerationStream({
40
  // @ts-ignore
41
+ model: 'meta-llama/Llama-2-70b-chat-hf',
42
  inputs: experimental_buildLlama2Prompt(messages),
43
  parameters: {
44
  max_new_tokens: 500,