Manglik-R commited on
Commit
91a1dfd
1 Parent(s): 9d41eb6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -50,8 +50,7 @@ model = Model(input_seq, output)
50
  model.compile(optimizer='adam',
51
  loss="sparse_categorical_crossentropy",
52
  metrics=["accuracy"])
53
- model.fit(x_train, y_train, epochs=20, batch_size=32,
54
- validation_data=(x_val, y_val))
55
  class Predict:
56
  def __init__(self, model, tokenizer):
57
  self.model = model
 
50
  model.compile(optimizer='adam',
51
  loss="sparse_categorical_crossentropy",
52
  metrics=["accuracy"])
53
+ model.fit(x_train, y_train, epochs=20, batch_size=32)
 
54
  class Predict:
55
  def __init__(self, model, tokenizer):
56
  self.model = model