Spaces:
Runtime error
Runtime error
import streamlit as st | |
from transformers import AutoTokenizer, AutoModelForCausalLM | |
def load_model(model_name): | |
with st.spinner('Waiting for the model to load.....'): | |
# snapshot_download('flax-community/Sinhala-gpt2') | |
tokenizer = AutoTokenizer.from_pretrained(model_name) | |
model = AutoModelForCausalLM.from_pretrained(model_name, pad_token_id=tokenizer.eos_token_id) | |
st.success('Model loaded!!') | |
return model, tokenizer |