add req.txt

app.py

@@ -19,10 +19,10 @@ def translate(src, tgt, text):
     # translate
     tokenizer.src_lang = src
     encoded_src = tokenizer(text, return_tensors="pt")
-    generated_tokens = model.generate(**encoded_src, forced_bos_token_id=tokenizer.get_lang_id(tgt))
-    result = tokenizer.batch_decode(generated_tokens, skip_special_tokens=True)
+    generated_tokens = model.generate(**encoded_src, forced_bos_token_id=tokenizer.get_lang_id(tgt), use_cache=True)
+    result = tokenizer.batch_decode(generated_tokens, skip_special_tokens=True)[0]
     return result
 
 output_text = gr.outputs.Textbox()
-gr.Interface(translate, inputs=[gr.inputs.Dropdown(lang_list, label="Source Language"), gr.inputs.Dropdown(lang_list, label="Target Language"), 'text'], outputs=output_text, title="M2M100",
-             description="100개국어 번역").launch()
+gr.Interface(translate, inputs=[gr.inputs.Dropdown(lang_list, label="Source Language"), gr.inputs.Dropdown(lang_list, label="Target Language"), 'text'], outputs=output_text, title="Translate Between 100 languages",
+             description="M2M100-1.2B 모델을 가지고 100개국어 언어를 번역하는 데모페이지 입니다.").launch()