wop commited on
Commit
c759268
1 Parent(s): 2cb8221

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -54,7 +54,7 @@ model = Sequential([
54
  model.compile(loss=sparse_categorical_crossentropy, optimizer='adam', metrics=['accuracy'])
55
 
56
  # Train model
57
- model.fit(question_sequences, answer_sequences, epochs=1000, batch_size=32, steps_per_epoch=1)
58
 
59
  # Function to generate response
60
  def generate_response(input_text):
 
54
  model.compile(loss=sparse_categorical_crossentropy, optimizer='adam', metrics=['accuracy'])
55
 
56
  # Train model
57
+ model.fit(question_sequences, answer_sequences, epochs=1000, batch_size=32, steps_per_epoch=2)
58
 
59
  # Function to generate response
60
  def generate_response(input_text):