gagan3012 / keytotext

Keywords to Sentences

Home Page:https://share.streamlit.io/gagan3012/keytotext/UI/app.py

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Local model problem

pendekarcode opened this issue · comments

I downloaded the model and saved it on a local PC. By code:

namefolder= 'mrm8488-t5-base'
tokenizer = AutoTokenizer.from_pretrained('mrm8488/t5-base-finetuned-common_gen')
model = AutoModelForSeq2SeqLM.from_pretrained('mrm8488/t5-base-finetuned-common_gen')
tokenizer.save_pretrained("./"+namefolder)
model.save_pretrained("./"+namefolder)

I can not run the model. Error:
site-packages\keytotext\pipeline.py", line 75, in pipeline
task, list(SUPPORTED_TASKS.keys())
KeyError: "Unknown task E:\PH\_MODEL_TRANSFORMERS\mrm8488-t5-base, available tasks are ['k2t', 'k2t-base', 'mrm8488/t5-base-finetuned-common_gen', 'k2t- new']"

Thanks