Hugging Face's logo

Spaces:
victor-johnson
/
emmet-generator
Running

App Files Community
victor-johnson commited on
Commit
ee67cd5
·
verified ·
1 Parent(s): 0f51816

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -6,7 +6,7 @@ from langchain_community.llms import HuggingFacePipeline
6
  from langchain import PromptTemplate, LLMChain
7
 
8
  # — Model setup (small enough to CPU-serve in a Space) —
9
- MODEL_ID = "bigcode/starcoder2-1b"
10
  tokenizer = AutoTokenizer.from_pretrained(MODEL_ID)
11
  model = AutoModelForCausalLM.from_pretrained(MODEL_ID)
12
 
 
6
  from langchain import PromptTemplate, LLMChain
7
 
8
  # — Model setup (small enough to CPU-serve in a Space) —
9
+ MODEL_ID = "bigcode/starcoder2-3b"
10
  tokenizer = AutoTokenizer.from_pretrained(MODEL_ID)
11
  model = AutoModelForCausalLM.from_pretrained(MODEL_ID)
12