Spaces:

kobkrit
/

openthaigpt

Paused

kobkrit commited on Feb 27, 2023

Commit

6b357e1

•

1 Parent(s): 2921564

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -5,7 +5,7 @@ import torch
 from torch.utils.data import Dataset, random_split
 from transformers import GPT2Tokenizer, TrainingArguments, Trainer, GPT2LMHeadModel
-pretrained_name = "kobkrit/openthaigpt-gpt2-pantipwiki-poc"
 tokenizer = GPT2Tokenizer.from_pretrained(pretrained_name, bos_token='<|startoftext|>',unk_token='<|unk|>', eos_token='<|endoftext|>', pad_token='<|pad|>')
 model = GPT2LMHeadModel.from_pretrained(pretrained_name).cuda()

 from torch.utils.data import Dataset, random_split
 from transformers import GPT2Tokenizer, TrainingArguments, Trainer, GPT2LMHeadModel
+pretrained_name = "kobkrit/openthaigpt-gpt2-instructgpt-poc-0.0.2"
 tokenizer = GPT2Tokenizer.from_pretrained(pretrained_name, bos_token='<|startoftext|>',unk_token='<|unk|>', eos_token='<|endoftext|>', pad_token='<|pad|>')
 model = GPT2LMHeadModel.from_pretrained(pretrained_name).cuda()