Update tr.py

tr.py

@@ -42,12 +42,12 @@ loss, accuracy = model.evaluate(X_test, y_test)
 print(f'Accuracy: {accuracy * 100:.2f}%')
 
 # Save the model with the correct filepath extension
-# model.save('my_custom_text_classifier')
-tf.saved_model.save(model, "my_custom_text_classifier")
+model.save('my_custom_text_classifier.keras')
+# tf.saved_model.save(model, "my_custom_text_classifier")
 
 # Later, load the model and make predictions
-# loaded_model = tf.keras.models.load_model('my_custom_text_classifier')
-loaded_model =tf.keras.layers.TFSMLayer("my_custom_text_classifier", call_endpoint="serving_default")
+loaded_model = tf.keras.models.load_model('my_custom_text_classifier.keras')
+# loaded_model =tf.keras.layers.TFSMLayer("my_custom_text_classifier", call_endpoint="serving_default")
 
 
 # Example prediction