modified arg

app.py

@@ -14,14 +14,13 @@ examples = [
     ["질문: 일본의 앞으로의 100년은 어떻게 될까요? \n답변:"],
     ["2040년 미국은, "]
 ]
-
 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-release", use_auth_token=os.environ['TOKEN']
 )
 model.eval()
 
-pipe = pipeline('text-generation', model=model, tokenizer=tokenizer)
+pipe = pipeline('text-generation', model=model, tokenizer=tokenizer, eos_token_id=tokenizer.eos_token_id)
 
 def predict(text):
   stopping_cond = StoppingCriteriaList([tokenizer.encode('<|endoftext|>')])