sentence_completion_gpt2

Runtime error

serhii-korobchenko commited on Jun 12, 2023

Commit

c280e3a

•

1 Parent(s): c120bb4

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -7,11 +7,11 @@ from tensorflow import keras
 import time
 def download_model_NLP():
-    #preprocessor = keras_nlp.models.GPT2CausalLMPreprocessor.from_preset(
-    #"gpt2_base_en",
-    #sequence_length=128,)
-    #model = keras_nlp.models.GPT2CausalLM.from_preset(
-    #"gpt2_base_en", preprocessor=preprocessor)
     #output = "GPT2_keras_deer"

 import time
 def download_model_NLP():
+    preprocessor = keras_nlp.models.GPT2CausalLMPreprocessor.from_preset(
+    "gpt2_base_en",
+    sequence_length=128,)
+    model = keras_nlp.models.GPT2CausalLM.from_preset(
+    "gpt2_base_en", preprocessor=preprocessor)
     #output = "GPT2_keras_deer"