Update app.py

app.py

@@ -15,7 +15,11 @@ lang_list = [lang.strip() for lang in langs.split(',')]
 model = M2M100ForConditionalGeneration.from_pretrained("alirezamsh/small100")
 tokenizer = SMALL100Tokenizer.from_pretrained("alirezamsh/small100")
 
-def small100_tr(text, lang):
+description = """This is a demo for the paper [*SMaLL-100: Introducing Shallow Multilingual Machine Translation Model for Low-Resource Languages*](https://arxiv.org/abs/2210.11621) by Alireza Mohammadshahi, Vassilina Nikoulina, Alexandre Berard, Caroline Brun, James Henderson, Laurent Besacier
+
+In this paper, they propose a compact and shallow massively multilingual MT model, and achieve competitive result with M2M-100, while being super smaller and faster."""
+
+def small100_tr(lang, text):
     tokenizer.tgt_lang = lang
     encoded_text = tokenizer(text, return_tensors="pt")
     generated_tokens = model.generate(**encoded_text)
@@ -23,11 +27,8 @@ def small100_tr(text, lang):
 
 examples = [["French (fr)", "엄마판다는 새끼가 있네."]]
 
-demo = gr.Interface(fn=small100_tr, inputs=["text", "text"], outputs="text")
-demo.launch()
-
 output_text = gr.outputs.Textbox()
 gr.Interface(small100_tr, inputs=[gr.inputs.Dropdown(lang_list, label=" Target Language"), 'text'], outputs=output_text, title="SMaLL100: Translate Between 100 languages much faster",
-            description="Demo page for SMaLL100 model",
+            description=description,
             examples=examples
             ).launch()