kobkrit commited on
Commit
9a6ad7e
1 Parent(s): 6b357e1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -5,7 +5,7 @@ import torch
5
  from torch.utils.data import Dataset, random_split
6
  from transformers import GPT2Tokenizer, TrainingArguments, Trainer, GPT2LMHeadModel
7
 
8
- pretrained_name = "kobkrit/openthaigpt-gpt2-instructgpt-poc-0.0.2"
9
 
10
  tokenizer = GPT2Tokenizer.from_pretrained(pretrained_name, bos_token='<|startoftext|>',unk_token='<|unk|>', eos_token='<|endoftext|>', pad_token='<|pad|>')
11
  model = GPT2LMHeadModel.from_pretrained(pretrained_name).cuda()
 
5
  from torch.utils.data import Dataset, random_split
6
  from transformers import GPT2Tokenizer, TrainingArguments, Trainer, GPT2LMHeadModel
7
 
8
+ pretrained_name = "kobkrit/openthaigpt-gpt2-instructgpt-poc-0.0.3"
9
 
10
  tokenizer = GPT2Tokenizer.from_pretrained(pretrained_name, bos_token='<|startoftext|>',unk_token='<|unk|>', eos_token='<|endoftext|>', pad_token='<|pad|>')
11
  model = GPT2LMHeadModel.from_pretrained(pretrained_name).cuda()