ignore model dir

.gitignore

@@ -0,0 +1 @@
+model

predict.py

@@ -0,0 +1,10 @@
+import torch
+
+
+def main():
+    vocab = model = torch.load('model/transformer.vocab.pt')
+    model = torch.load('model/model_step_15000.pt')
+    print(vocab)
+
+if __name__ == "__main__":
+    main()