Add script

.gitattributes

@@ -32,3 +32,4 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
 *.zip filter=lfs diff=lfs merge=lfs -text
 *.zst filter=lfs diff=lfs merge=lfs -text
 *tfevents* filter=lfs diff=lfs merge=lfs -text
+model filter=lfs diff=lfs merge=lfs -text

app.py

@@ -0,0 +1,49 @@
+import gradio as gr
+import tensorflow as tf
+import keras_nlp
+
+MAX_SEQUENCE_LENGTH = 40
+model = tf.keras.models.load_model('model')
+
+eng_tokenizer = keras_nlp.tokenizers.SentencePieceTokenizer(
+            'spmodel/m48.model')
+tha_tokenizer = keras_nlp.tokenizers.SentencePieceTokenizer(
+            'spmodel/m48.model')
+
+def decode_sequences(input_sentences):
+    batch_size = tf.shape(input_sentences)[0]
+
+    encoder_input_tokens = eng_tokenizer(input_sentences).to_tensor(
+        shape=(None, MAX_SEQUENCE_LENGTH)
+    )
+    def token_probability_fn(decoder_input_tokens):
+        return model([encoder_input_tokens, decoder_input_tokens])[:, -1, :]
+
+    prompt = tf.fill((batch_size, 1), tha_tokenizer.token_to_id("<s>"))
+
+    generated_tokens = keras_nlp.utils.top_p_search(
+        token_probability_fn,
+        prompt,
+        p=0.1,
+        max_length=40,
+        end_token_id=tha_tokenizer.token_to_id("</s>"),
+    )
+
+    generated_sentences = tha_tokenizer.detokenize(generated_tokens)
+    return generated_sentences
+
+def greet(english_lyrics):
+    translated = decode_sequences(tf.constant([english_lyrics]))
+    translated = translated.numpy()[0].decode("utf-8")
+    translated = (
+        translated
+        .replace("<pad>", "")
+        .replace("<s>", "")
+        .replace("</s>", "")
+        .replace("⁇", "")
+        .strip()
+    )
+    return translated
+
+iface = gr.Interface(fn=greet, inputs="text", outputs="text")
+iface.launch()

requirements.txt

@@ -0,0 +1,3 @@
+gradio
+tensorflow
+keras_nlp

saved_model/model/fingerprint.pb

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:02779466cdec163811d078815c633f21901413081449002f24aa3e80f0b88ef7
+size 55

saved_model/model/keras_metadata.pb

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:e7217045d222bb77b6906d09ec5aa81f8d008171925996a3f515fcf3e58acff7
+size 58966

saved_model/model/saved_model.pb

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:aa4d4866561073f0aed3750adf47ca690fbfffc4489010b91068b2104784ad74
+size 1419373

saved_model/model/variables/variables.data-00000-of-00001

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:915a0ce9d50ac6b8d17e2aae56407c975936e4e1f66b53d63922c4d5e057b16d
+size 52920719

saved_model/model/variables/variables.index

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:cd3a16ac228e0766ebc1db8b4bb91b59117f1e0612961bff706fb3550e916083
+size 6120

spmodel/m48.model

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:e7f360c04636a14301b9d5623e083bd553d1dda137e6edf4988e98b2a6344cf4
+size 354171