Update main.py

main.py

@@ -9,6 +9,11 @@ from tensorflow.keras.preprocessing.sequence import pad_sequences
 from tensorflow.keras.preprocessing.text import Tokenizer
 from tensorflow.keras.utils import to_categorical
 from tensorflow import keras
+from datasets import load_dataset
+
+# Login using e.g. `huggingface-cli login` to access this dataset
+ds = load_dataset("GeneralReasoning/GeneralThought-323K")
+
 sentences = [
 'Life is so beautiful',
 'Hope keeps us going',