HumamZaman's picture
Update app.py
ca0b363 verified
raw
history blame contribute delete
338 Bytes
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)