SayknowLab commited on
Commit
5bb4640
·
verified ·
1 Parent(s): cbe2a47

Update app.py

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