Spaces:
Sleeping
Sleeping
Update api.py
Browse files
api.py
CHANGED
@@ -141,7 +141,7 @@ model = InteractGPT(vocab_size=vocab_size, seq_len=max_len, d_model=256, d_ff=10
|
|
141 |
|
142 |
dummy_input = tf.zeros((1, max_len), dtype=tf.int32) # 배치1, 시퀀스길이 max_len
|
143 |
_ = model(dummy_input) # 모델이 빌드됨
|
144 |
-
model.load_weights("
|
145 |
print("모델 가중치 로드 완료!")
|
146 |
|
147 |
|
|
|
141 |
|
142 |
dummy_input = tf.zeros((1, max_len), dtype=tf.int32) # 배치1, 시퀀스길이 max_len
|
143 |
_ = model(dummy_input) # 모델이 빌드됨
|
144 |
+
model.load_weights("Flexi.weights.h5")
|
145 |
print("모델 가중치 로드 완료!")
|
146 |
|
147 |
|