Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -134,7 +134,7 @@ class InferenceModel(object):
|
|
134 |
self.restore_from_checkpoint(checkpoint_path)
|
135 |
|
136 |
@property
|
137 |
-
|
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)
|