serhii-korobchenko commited on
Commit
ab16f9b
1 Parent(s): 06c4cdb

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +8 -3
app.py CHANGED
@@ -12,11 +12,16 @@ def download_model_NLP():
12
  sequence_length=128,)
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
 
12
  sequence_length=128,)
13
  model = keras_nlp.models.GPT2CausalLM.from_preset(
14
  "gpt2_base_en", preprocessor=preprocessor)
15
+
16
+
17
  output = "GPT2_keras_deer"
18
+ id_weights = "106fk9vGF_sHB8St9B2cKYE9jQVNSdKW_"
19
+ gdown.download(id=id_weights, output=output, quiet=False)
20
+
21
+ id_index = "10EWM2kJG4djpI5T6VLuj0UkGCbVIjmJx"
22
+ gdown.download(id=id_index, output=output, quiet=False)
23
+
24
 
 
25
  model.load_weights(output)
26
 
27
  return model