Titanicc73 commited on
Commit
c5df11c
·
1 Parent(s): 313e391

Add custom inference script and model files

Browse files
Files changed (4) hide show
  1. app.py +31 -0
  2. lstm_model.h5 +3 -0
  3. padding_config.pkl +3 -0
  4. 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