Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
|
@@ -12,9 +12,9 @@ app = Flask(__name__)
|
|
| 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. 데이터셋 로드 (기존과 동일)
|
|
|
|
| 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. 데이터셋 로드 (기존과 동일)
|