tigeryfan commited on
Commit
26a5fd4
·
verified ·
1 Parent(s): 42dba8e

Update backend.py

Browse files
Files changed (1) hide show
  1. backend.py +2 -2
backend.py CHANGED
@@ -2,9 +2,9 @@ from transformers import pipeline
2
  from transformers import AutoTokenizer, AutoModelForCausalLM, pipeline
3
 
4
  def load_model():
5
- tokenizer = AutoTokenizer.from_pretrained("openai/gpt-oss-20b")
6
  model = AutoModelForCausalLM.from_pretrained("openai/gpt-oss-20b")
7
- return pipeline("text2text-generation", model=model, tokenizer=tokenizer)
8
 
9
  pipe = load_model()
10
 
 
2
  from transformers import AutoTokenizer, AutoModelForCausalLM, pipeline
3
 
4
  def load_model():
5
+ tokenizer = AutoTokenizer.from_pretrained("openai/gpt-oss-20b")
6
  model = AutoModelForCausalLM.from_pretrained("openai/gpt-oss-20b")
7
+ return pipeline("text2text-generation", model=model, tokenizer=tokenizer)
8
 
9
  pipe = load_model()
10