Update app.py

app.py

@@ -20,12 +20,15 @@ description = """This is a demo for the paper [*SMaLL-100: Introducing Shallow M
 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):
+
+    lang = lang.split(" ")[-1][1:-1]
+    
     tokenizer.tgt_lang = lang
     encoded_text = tokenizer(text, return_tensors="pt")
     generated_tokens = model.generate(**encoded_text)
     return tokenizer.batch_decode(generated_tokens, skip_special_tokens=True)[0]
 
-examples = [["French (fr)", "엄마판다는 새끼가 있네."]]
+examples = [["French (fr)", "Life is like a box of chocolates."]]
 
 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",