Update app.py

app.py

@@ -1,6 +1,7 @@
-import gradio as gr
+import gradio
 from transformers import AutoModelForSeq2SeqLM, AutoTokenizer
 
+
 model_name = "zaanind/nllb-ensi-v1-tuning"
 tokenizer = AutoTokenizer.from_pretrained(model_name)
 model = AutoModelForSeq2SeqLM.from_pretrained(model_name)
@@ -15,10 +16,17 @@ def translate(text):
     translation = tokenizer.decode(translated_tokens[0], skip_special_tokens=True)
     return translation
 
-iface = gr.Interface(fn=translate, 
-                     inputs="text", 
-                     outputs="text",
-                     title="EnSi NMT",
-                     description="This is a demo and rest api for ensi nmt by @zaanind."
-                    )
-iface.launch()
+    
+def nmtapifunc(text):
+    text = translate(text)
+    return text 
+
+gradio_interface = gradio.Interface(
+  fn=nmtapifunc,
+  inputs="text",
+  outputs="text",
+  title="En-Si NMT",
+  description="",
+  article="© zaanind 2024"
+)
+gradio_interface.launch()