Update app.py

app.py

@@ -1,3 +1,5 @@
+# attribution: code for demo is based on https://huggingface.co/spaces/Geonmo/nllb-translation-demo
+
 from fastapi import FastAPI, Depends, HTTPException, Request
 from fastapi.security import APIKeyQuery
 from pydantic import BaseModel
@@ -114,6 +116,34 @@ async def logo():
 
 
 io = gr.Interface(lambda x: "Hello, " + x + "!", "textbox", "textbox")
+lang_codes = list(flores_codes.keys())
+#inputs = [gr.inputs.Radio(['nllb-distilled-600M', 'nllb-1.3B', 'nllb-distilled-1.3B'], label='NLLB Model'),
+inputs = [gr.Dropdown(lang_codes, value='English', label='Source'),
+          gr.Dropdown(lang_codes, value='Crimean Tatar', label='Target'),
+          gr.Textbox(lines=5, label="Input text"),
+          ]
+
+outputs = gr.JSON()
+
+title = "Crimean Tatar Translator based on NLLB distilled 600M demo"
+
+
+description = f"Details: https://github.com/facebookresearch/fairseq/tree/nllb."
+examples = [
+['English', 'Korean', 'Hi. nice to meet you']
+]
+
+def translate_single(source_lang: str, target_lang: str, text: str) -> List[str]:):
+    return translate_text([text], source_lang, target_lang)[0]
+
+gr.Interface(translate_single,
+             inputs,
+             outputs,
+             title=title,
+             description=description,
+             )
+
+
 app = gr.mount_gradio_app(app, io, path=CUSTOM_PATH)
 
 if __name__ == "__main__":