danhtran2mind
/

En2Vi-Translation-Transformer-TensorFlow

Model card Files Files and versions

danhtran2mind commited on Nov 11

Commit

9e17272

·

verified ·

1 Parent(s): 111b2ca

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -56,7 +56,7 @@ if __name__ == "__main__":
     }
     # Load the model
-    loaded_model = tf.keras.models.load_model('saved_models/en_vi_translation.keras',
                                             custom_objects=custom_objects)
     main(sentence=sentence, model=loaded_model)

     }
     # Load the model
+    loaded_model = tf.keras.models.load_model('ckpts/en_vi_translation.keras',
                                             custom_objects=custom_objects)
     main(sentence=sentence, model=loaded_model)