Update tf_chatbot.py

Deadsg · Nov 19, 2023 · cff1ff1 · cff1ff1
1 parent df037fa
commit cff1ff1
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/CHATBOT/tf_chatbot.py b/CHATBOT/tf_chatbot.py
@@ -42,15 +42,15 @@
     tf.keras.layers.Embedding(input_dim=128, output_dim=128, input_length=50),
     tf.keras.layers.LSTM(units=128, return_sequences=True),
     tf.keras.layers.LSTM(units=128),
-    tf.keras.layers.Dense(128, activation='softmax')  
+    tf.keras.layers.Dense(128, activation='softmax') 
 ])
-seq2seq_model.compile(optimizer='adam', loss='sparse_categorical_crossentropy', metrics=['accuracy'])
+seq2seq_model.compile(
+    optimizer='adam', loss='sparse_categorical_crossentropy', metrics=['accuracy'])
 
 # CLI Chatbot
 while True:
     user_input = input("You: ")
 
-    # Exit the chatbot if the user enters 'exit'
     if user_input.lower() == 'exit':
         print("Chatbot: Goodbye!")
         break