Spaces:

MoyAI
/

Relevance-classifier

Sleeping

ierhon commited on Sep 2, 2023

Commit

7e24ae9

1 Parent(s): 7fb5ad9

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -3,7 +3,7 @@ import pickle
 from keras.models import load_model
 from keras_self_attention import SeqSelfAttention
-model = load_model("model.keras", custom_objects:{"SeqSelfAttention":SeqSelfAttention})
 with open("tokenizer.pckl", "rb") as file:
     tokenizer = pickle.load(file)

 from keras.models import load_model
 from keras_self_attention import SeqSelfAttention
+model = load_model("model.keras", custom_objects={"SeqSelfAttention":SeqSelfAttention})
 with open("tokenizer.pckl", "rb") as file:
     tokenizer = pickle.load(file)