Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
|
@@ -11,9 +11,9 @@ app = Flask(__name__)
|
|
| 11 |
|
| 12 |
# --- 1. 모델 로드 ---
|
| 13 |
print("토크나이저 로딩 중...")
|
| 14 |
-
tokenizer = AutoTokenizer.from_pretrained("
|
| 15 |
print("모델 로딩 중...")
|
| 16 |
-
model = GPT2LMHeadModel.from_pretrained("
|
| 17 |
print("모델 로딩 완료!")
|
| 18 |
|
| 19 |
# --- 2. 데이터셋 로드 ---
|
|
|
|
| 11 |
|
| 12 |
# --- 1. 모델 로드 ---
|
| 13 |
print("토크나이저 로딩 중...")
|
| 14 |
+
tokenizer = AutoTokenizer.from_pretrained("Nanbeige/Nanbeige4.1-3B", trust_remote_code=True)
|
| 15 |
print("모델 로딩 중...")
|
| 16 |
+
model = GPT2LMHeadModel.from_pretrained("Nanbeige/Nanbeige4.1-3B", trust_remote_code=True)
|
| 17 |
print("모델 로딩 완료!")
|
| 18 |
|
| 19 |
# --- 2. 데이터셋 로드 ---
|