File size: 338 Bytes
ca0b363
 
cb3cff6
cd496f2
ca0b363
cb3cff6
ca0b363
 
1
2
3
4
5
6
7
8
9
import os
from transformers import AutoTokenizer, AutoModelForCausalLM

model_name = "openai-community/gpt-oss-20b"
hf_token = os.getenv("HF_TOKEN")  # Reads from Space secret

tokenizer = AutoTokenizer.from_pretrained(model_name, use_auth_token=hf_token)
model = AutoModelForCausalLM.from_pretrained(model_name, use_auth_token=hf_token)