Spaces:
Runtime error
Runtime error
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)
|