Spaces:
Runtime error
Runtime error
Commit
·
06c4cdb
1
Parent(s):
4e59545
Update app.py
Browse files
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 = "
|
17 |
-
|
18 |
-
|
|
|
|
|
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 |
|