Spaces:
Runtime error
Runtime error
Commit
·
054d223
1
Parent(s):
0366d5b
Create app.py
Browse files
app.py
ADDED
@@ -0,0 +1,25 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
import gradio as gr
|
2 |
+
import tensorflow as tf
|
3 |
+
import numpy as np
|
4 |
+
import joblib
|
5 |
+
from tensorflow.keras.preprocessing.text import Tokenizer
|
6 |
+
from tensorflow.keras.preprocessing.sequence import pad_sequences
|
7 |
+
|
8 |
+
model = tf.keras.models.load_model("best_model.h5")
|
9 |
+
english_tokenizer = joblib.load('english_tokenizer')
|
10 |
+
hindi_tokenizer = joblib.load('hindi_tokenizer')
|
11 |
+
|
12 |
+
def generate_text(txts):
|
13 |
+
txts = txts.split(' ')
|
14 |
+
res = []
|
15 |
+
for txt in txts:
|
16 |
+
tmp_tok = english_tokenizer.texts_to_sequences([txt])
|
17 |
+
tmp_pad = pad_sequences(tmp_tok, padding='post', maxlen=100, truncating='post')
|
18 |
+
tmp_preds = model.predict(tmp_pad,verbose=0)
|
19 |
+
text2 = hindi_tokenizer.sequences_to_texts([[np.argmax(p)] for p in tmp_preds[0]])
|
20 |
+
res.append("".join(text2))
|
21 |
+
return " ".join(res)
|
22 |
+
|
23 |
+
demo = gr.Interface(fn=generate_text, inputs="text", outputs="text")
|
24 |
+
|
25 |
+
demo.launch(share=True)
|