Update app.py

app.py

@@ -28,17 +28,17 @@ import keras_nlp
 import time
 import re
 
-os.environ["KERAS_BACKEND"] = "tensorflow"  # or "tensorflow" or "torch"
+#os.environ["KERAS_BACKEND"] = "tensorflow"  # or "tensorflow" or "torch"
 
-keras.mixed_precision.set_global_policy("mixed_float16")
+#keras.mixed_precision.set_global_policy("mixed_float16")
 
-preprocessor = keras_nlp.models.GPT2CausalLMPreprocessor.from_preset(
-    "gpt2_medium_en",
-    sequence_length=128,
-)
-gpt2_lm = keras_nlp.models.GPT2CausalLM.from_preset(
-    "gpt2_medium_en", preprocessor=preprocessor
-)
+#preprocessor = keras_nlp.models.GPT2CausalLMPreprocessor.from_preset(
+#    "gpt2_medium_en",
+#    sequence_length=128,
+#)
+#gpt2_lm = keras_nlp.models.GPT2CausalLM.from_preset(
+#    "gpt2_medium_en", preprocessor=preprocessor
+#)
 
 #drive.mount('/content/drive', force_remount=True)
 # GPT2 Base 1 epochs
@@ -46,9 +46,11 @@ gpt2_lm = keras_nlp.models.GPT2CausalLM.from_preset(
 # GPT2 Base 3 epochs
 #checkpoint_path = "./aloqas_model_checkpoints_gpt2base_3_epochs/cp.ckpt"
 # GPT2 Base 3 epochs
-checkpoint_path = "./aloqas_model_checkpoints_gpt2medium_3_epochs/cp.ckpt"
+#checkpoint_path = "./aloqas_model_checkpoints_gpt2medium_3_epochs/cp.ckpt"
 
-gpt2_lm.load_weights(checkpoint_path)
+#gpt2_lm.load_weights(checkpoint_path)
+
+gpt2_lm = keras.models.load_model("aloqas_model_keras_gpt2medium_3epochs/aloqas_model_gpt2medium_3_epochs.keras")
 
 """# Chargement et configuration de gradio"""