serhii-korobchenko commited on
Commit
06c4cdb
·
1 Parent(s): 4e59545

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -3
app.py CHANGED
@@ -13,9 +13,11 @@ def download_model_NLP():
13
  model = keras_nlp.models.GPT2CausalLM.from_preset(
14
  "gpt2_base_en", preprocessor=preprocessor)
15
 
16
- id = "1zi3hSBRTP9uwHDVwQMqpGeXb8RBhJfrO"
17
- model_path = gdown.download_folder(id=id, quiet=True, use_cookies=False)
18
- model.load_weights(model_path)
 
 
19
 
20
  return model
21
 
 
13
  model = keras_nlp.models.GPT2CausalLM.from_preset(
14
  "gpt2_base_en", preprocessor=preprocessor)
15
 
16
+ id = "106fk9vGF_sHB8St9B2cKYE9jQVNSdKW_"
17
+ output = "GPT2_keras_deer"
18
+
19
+ gdown.download(id=id, output=output, quiet=False)
20
+ model.load_weights(output)
21
 
22
  return model
23