Spaces:

MoyAI
/

Relevance-classifier

Sleeping

ierhon commited on Sep 2, 2023

Commit

7fb5ad9

1 Parent(s): c1edad1

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -1,8 +1,9 @@
 import gradio as gr
 import pickle
 from keras.models import load_model
-model = load_model("model.keras")
 with open("tokenizer.pckl", "rb") as file:
     tokenizer = pickle.load(file)

 import gradio as gr
 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)