Spaces:
Sleeping
Sleeping
Commit
·
c5df11c
1
Parent(s):
313e391
Add custom inference script and model files
Browse files- app.py +31 -0
- lstm_model.h5 +3 -0
- padding_config.pkl +3 -0
- tokenizer.pkl +3 -0
app.py
ADDED
@@ -0,0 +1,31 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
import gradio as gr
|
2 |
+
import tensorflow as tf
|
3 |
+
import joblib
|
4 |
+
import pickle
|
5 |
+
from tensorflow.keras.preprocessing.sequence import pad_sequences
|
6 |
+
|
7 |
+
# Load your model and tokenizer
|
8 |
+
model = tf.keras.models.load_model('lstm_model.h5')
|
9 |
+
tokenizer = joblib.load('tokenizer.pkl')
|
10 |
+
|
11 |
+
with open('padding_config.pkl', 'rb') as file:
|
12 |
+
padding_config = pickle.load(file)
|
13 |
+
|
14 |
+
# Preprocessing function
|
15 |
+
def preprocess(text):
|
16 |
+
tokenized_text = tokenizer.texts_to_sequences([text])
|
17 |
+
padded_text = pad_sequences(tokenized_text, **padding_config)
|
18 |
+
return padded_text
|
19 |
+
|
20 |
+
# Prediction function
|
21 |
+
def predict(text):
|
22 |
+
processed_text = preprocess(text)
|
23 |
+
prediction = model.predict(processed_text)
|
24 |
+
return prediction.tolist()
|
25 |
+
|
26 |
+
# Gradio interface
|
27 |
+
iface = gr.Interface(fn=predict, inputs="text", outputs="json")
|
28 |
+
|
29 |
+
# Launch Gradio app
|
30 |
+
iface.launch()
|
31 |
+
|
lstm_model.h5
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:4c0a6a4c3b2b4e04b5b4969f702aa9e3658807876ea14aa645929f0bdfed8622
|
3 |
+
size 3047376
|
padding_config.pkl
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:a657ebd6b118ff15b47ff05d0cbad88200afbb206a424531eba767c3641bd01e
|
3 |
+
size 44
|
tokenizer.pkl
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:92911129cda7e87c410497dd5d3a26307dcbd34c107cc0d422ffcd538687461e
|
3 |
+
size 10093140
|