Sarah Ciston commited on
Commit
addde8c
·
1 Parent(s): 5b326aa

update models loaded

Browse files
Files changed (2) hide show
  1. README.md +3 -2
  2. sketch.js +1 -1
README.md CHANGED
@@ -7,8 +7,9 @@ sdk: static
7
  pinned: false
8
  models:
9
  - Xenova/detr-resnet-50
10
- - mistralai/Mistral-7B-Instruct-v0.2
11
- - meta-llama/Meta-Llama-3-8B
 
12
  ---
13
 
14
  Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference
 
7
  pinned: false
8
  models:
9
  - Xenova/detr-resnet-50
10
+ - Xenova/gpt2
11
+ # - mistralai/Mistral-7B-Instruct-v0.2
12
+ # - meta-llama/Meta-Llama-3-8B
13
  ---
14
 
15
  Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference
sketch.js CHANGED
@@ -4,7 +4,7 @@ import { pipeline, env } from 'https://cdn.jsdelivr.net/npm/@xenova/transformers
4
 
5
  // import { pipeline } from '@xenova/transformers';
6
 
7
- let pipe = await pipeline('text-generation', 'mistralai/Mistral-7B-Instruct-v0.2');
8
  // models('Xenova/gpt2', 'mistralai/Mistral-7B-Instruct-v0.2', 'meta-llama/Meta-Llama-3-8B')
9
 
10
  // Since we will download the model from the Hugging Face Hub, we can skip the local model check
 
4
 
5
  // import { pipeline } from '@xenova/transformers';
6
 
7
+ let pipe = await pipeline('text-generation', 'Xenova/gpt2');
8
  // models('Xenova/gpt2', 'mistralai/Mistral-7B-Instruct-v0.2', 'meta-llama/Meta-Llama-3-8B')
9
 
10
  // Since we will download the model from the Hugging Face Hub, we can skip the local model check