Update tr.py

tr.py

@@ -36,11 +36,11 @@ model.fit(X_train, y_train, epochs=5, batch_size=2)
 loss, accuracy = model.evaluate(X_test, y_test)
 print(f'Accuracy: {accuracy * 100:.2f}%')
 
-# Save the model
-model.save('my_custom_text_classifier')
+# Save the model with the correct filepath extension
+model.save('my_custom_text_classifier.h5')
 
 # Later, load the model and make predictions
-loaded_model = tf.keras.models.load_model('my_custom_text_classifier')
+loaded_model = tf.keras.models.load_model('my_custom_text_classifier.h5')
 
 # Example prediction
 new_texts = ["I'm feeling great!", "This book is boring."]