Fix lang

app.py

@@ -14,8 +14,7 @@ device = 'cuda' if torch.cuda.is_available() else 'cpu'
 
 
 def summarize(lang, text):
-    tokenizer = RobertaTokenizerFast.from_pretrained(models_paths[lang]) if lang in [
-        "fr", "es"] else BertTokenizerFast.from_pretrained(models_paths[lang])
+    tokenizer = RobertaTokenizerFast.from_pretrained(models_paths[lang]) if lang == "fr" or lang == "es" else BertTokenizerFast.from_pretrained(models_paths[lang])
     model = EncoderDecoderModel.from_pretrained(models_paths[lang]).to(device)
     inputs = tokenizer([text], padding="max_length",
                        truncation=True, max_length=512, return_tensors="pt")
@@ -25,5 +24,5 @@ def summarize(lang, text):
     return tokenizer.decode(output[0], skip_special_tokens=True)
 
 
-gr.Interface(fn=summarize, inputs=[gr.inputs.CheckboxGroup(["fr", "de", "tu", "es"]), gr.inputs.Textbox(
+gr.Interface(fn=summarize, inputs=[gr.inputs.Radio(["fr", "de", "tu", "es"]), gr.inputs.Textbox(
     lines=7, label="Input Text")], outputs="text").launch(inline=False)