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 |