alexandre-huynh commited on
Commit
a91e32c
1 Parent(s): 6cb7c4c

Updated app.py to use new finetuned gpt2 medium model

Browse files
Files changed (1) hide show
  1. app.py +4 -2
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