llmademo / app.py
Shankarm08's picture
Update app.py
fba67f5 verified
import hf_api
# Authenticate with Hugging Face
hf_api.set_access_token(None) # Use the default access token
# Load model directly
from transformers import AutoTokenizer, AutoModelForCausalLM
tokenizer = AutoTokenizer.from_pretrained("meta-llama/Meta-Llama-3-8B")
model = AutoModelForCausalLM.from_pretrained("meta-llama/Meta-Llama-3-8B")