ierhon commited on
Commit
4b443ba
·
1 Parent(s): 676a08f

Create test.py

Browse files
Files changed (1) hide show
  1. test.py +7 -0
test.py ADDED
@@ -0,0 +1,7 @@
 
 
 
 
 
 
 
 
1
+ import json # TODO: use the responses file after it's done instead of loading the dataset again
2
+ import numpy as np
3
+ from keras.saving import load_model
4
+ from keras_self_attention import SeqSelfAttention
5
+ from vecs import *
6
+
7
+ model = load_model("model.keras", custom_objects={"SeqSelfAttention": SeqSelfAttention})