Hmjz100 commited on
Commit
7bd3414
1 Parent(s): 1db8186

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -134,7 +134,7 @@ class InferenceModel(object):
134
  self.restore_from_checkpoint(checkpoint_path)
135
 
136
  @property
137
- def input_shapes(self):
138
  return {
139
  'encoder_input_tokens': (self.batch_size, self.inputs_length),
140
  'decoder_input_tokens': (self.batch_size, self.outputs_length)
 
134
  self.restore_from_checkpoint(checkpoint_path)
135
 
136
  @property
137
+ def input_shapes(self):
138
  return {
139
  'encoder_input_tokens': (self.batch_size, self.inputs_length),
140
  'decoder_input_tokens': (self.batch_size, self.outputs_length)