app.py

import os
import gradio as gr
import tensorflow as tf
import tensorflow_text as tf_text
from huggingface_hub import Repository

repo = Repository(
	local_dir="nmt-bahdanau-attention",
	clone_from="pyimagesearch/nmt-bahdanau-attention",
	use_auth_token=os.environ.get("token")
)
reloaded = tf.saved_model.load("nmt-bahdanau-attention/translator")

def get_translation(sentence):
	result = reloaded.translate(
		sourceText=tf.constant([sentence])
	)["text"].numpy()[0].decode()
	return result

nmt_space = gr.Interface(
	fn=get_translation,
	inputs="text",
	outputs="text"
)

nmt_space.launch()