Update to version 0.0.4

app.py

@@ -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-instructgpt-poc-0.0.3"
+pretrained_name = "kobkrit/openthaigpt-gpt2-instructgpt-poc-0.0.4"
 
 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()
@@ -18,4 +18,4 @@ def gen(input):
     return tokenizer.decode(output[0], skip_special_tokens=True)
 
 demo = gr.Interface(fn=gen, inputs=gr.Textbox(lines=3, label="Input Text", value="Q: อยากลดความอ้วน ทำอย่างไร\n\nA:"), outputs="text")
-demo.launch()   
+demo.launch()