article-to-sog / app.py
kitty528's picture
Update app.py
fc62a49 verified
raw
history blame contribute delete
280 Bytes
from transformers import AutoModelForCausalLM, AutoTokenizer
model_name = "kitty528/article-to-song-final"
# Load the model and tokenizer
model = AutoModelForCausalLM.from_pretrained(model_name,ignore_mismatched_sizes=True)
tokenizer = AutoTokenizer.from_pretrained(model_name)