Spaces:
Sleeping
Sleeping
Update api.py
Browse files
api.py
CHANGED
@@ -139,8 +139,8 @@ model.load_weights("InteractGPT.weights.h5")
|
|
139 |
print("모델 가중치 로드 완료!")
|
140 |
|
141 |
def generate_text_top_kp(model, prompt, max_len=100, max_gen=98,
|
142 |
-
temperature=
|
143 |
-
repetition_penalty=1.1, top_k=
|
144 |
model_input = text_to_ids(f"<start> {prompt} <sep>")
|
145 |
model_input = model_input[:max_len]
|
146 |
generated = list(model_input)
|
|
|
139 |
print("모델 가중치 로드 완료!")
|
140 |
|
141 |
def generate_text_top_kp(model, prompt, max_len=100, max_gen=98,
|
142 |
+
temperature=0.7, min_len=20,
|
143 |
+
repetition_penalty=1.1, top_k=50, top_p=0.9):
|
144 |
model_input = text_to_ids(f"<start> {prompt} <sep>")
|
145 |
model_input = model_input[:max_len]
|
146 |
generated = list(model_input)
|