Spaces:
Runtime error
Runtime error
alexandre-huynh
commited on
Commit
•
a91e32c
1
Parent(s):
6cb7c4c
Updated app.py to use new finetuned gpt2 medium model
Browse files
app.py
CHANGED
@@ -42,9 +42,11 @@ gpt2_lm = keras_nlp.models.GPT2CausalLM.from_preset(
|
|
42 |
|
43 |
#drive.mount('/content/drive', force_remount=True)
|
44 |
# GPT2 Base 1 epochs
|
45 |
-
checkpoint_path = "./aloqas_model_checkpoints/cp.ckpt"
|
46 |
# GPT2 Base 3 epochs
|
47 |
-
checkpoint_path = "./aloqas_model_checkpoints_gpt2base_3_epochs/cp.ckpt"
|
|
|
|
|
48 |
|
49 |
gpt2_lm.load_weights(checkpoint_path)
|
50 |
|
|
|
42 |
|
43 |
#drive.mount('/content/drive', force_remount=True)
|
44 |
# GPT2 Base 1 epochs
|
45 |
+
#checkpoint_path = "./aloqas_model_checkpoints/cp.ckpt"
|
46 |
# GPT2 Base 3 epochs
|
47 |
+
#checkpoint_path = "./aloqas_model_checkpoints_gpt2base_3_epochs/cp.ckpt"
|
48 |
+
# GPT2 Base 3 epochs
|
49 |
+
checkpoint_path = "./aloqas_model_checkpoints_gpt2medium_3_epochs/cp.ckpt"
|
50 |
|
51 |
gpt2_lm.load_weights(checkpoint_path)
|
52 |
|