modified target

app.py

@@ -11,16 +11,13 @@ description = "GPT-NeoX 한국어 모델을 시연하는 데모페이지 입니
 # article = "<p style='text-align: center'><a href='https://github.com/kingoflolz/mesh-transformer-jax' target='_blank'>GPT-J-6B: A 6 Billion Parameter Autoregressive Language Model</a></p>"
 examples = [
     ["인간처럼 생각하고, 행동하는 '지능'을 통해"],
-    ["질문: 한가위를 가장 의미있게 보낼수 있는 방법은 무엇일까요? \n답변:"],
+    ["질문: 일본의 앞으로의 100년은 어떻게 될까요? \n답변:"],
     ["2040년 미국은, "]
 ]
 
-tokenizer = AutoTokenizer.from_pretrained("EleutherAI/gpt-neox-ko-1.3b", use_auth_token=os.environ['TOKEN'])
+tokenizer = AutoTokenizer.from_pretrained("EleutherAI/gpt-neox-ko-1.3b-release", use_auth_token=os.environ['TOKEN'])
 model = AutoModelForCausalLM.from_pretrained(
-  "EleutherAI/gpt-neox-ko-1.3b", use_auth_token=os.environ['TOKEN'],
-  pad_token_id=tokenizer.pad_token_id,
-  eos_token_id=tokenizer.eos_token_id,
-  bos_token_id=tokenizer.bos_token_id  
+  "EleutherAI/gpt-neox-ko-1.3b-release", use_auth_token=os.environ['TOKEN']
 )
 model.eval()