Spaces:

flax-community
/

SinhalaLanguageDemos

Runtime error

changing some styles

ca404cd over 5 years ago

456 Bytes

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

	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