SayknowLab commited on
Commit
9dfccb8
·
verified ·
1 Parent(s): 5d78baa

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -12,9 +12,9 @@ app = Flask(__name__)
12
 
13
  # 1. 모델 로드 (기존과 동일)
14
  print("토크나이저 로딩 중...")
15
- tokenizer = AutoTokenizer.from_pretrained("skt/kogpt2-base-v2", trust_remote_code=True)
16
  print("모델 로딩 중...")
17
- model = GPT2LMHeadModel.from_pretrained("skt/kogpt2-base-v2", trust_remote_code=True)
18
  print("모델 로딩 완료!")
19
 
20
  # 2. 데이터셋 로드 (기존과 동일)
 
12
 
13
  # 1. 모델 로드 (기존과 동일)
14
  print("토크나이저 로딩 중...")
15
+ tokenizer = AutoTokenizer.from_pretrained("skt/kogpt2", trust_remote_code=True)
16
  print("모델 로딩 중...")
17
+ model = GPT2LMHeadModel.from_pretrained("skt/kogpt2", trust_remote_code=True)
18
  print("모델 로딩 완료!")
19
 
20
  # 2. 데이터셋 로드 (기존과 동일)