Update app.py

app.py

@@ -207,7 +207,7 @@ def beam_search_decode(model, src, src_mask, max_len, start_symbol, beam_size ,l
     best_beam = beams[0][0]
     return best_beam
 
-def translate(model: torch.nn.Module, src_sentence: str, strategy:str = 'greedy' , lenght_extend :int = 5, beam_size: int = 5,  length_penalty:float = 0.6):
+def translate(src_sentence: str, strategy:str = 'greedy' , lenght_extend :int = 5, beam_size: int = 5,  length_penalty:float = 0.6):
     assert strategy in ['greedy','beam search'], 'the strategy for decoding has to be either greedy or beam search'
     # Tokenize the source sentence
     src = source_tokenizer(src_sentence, **token_config)['input_ids']
@@ -259,7 +259,7 @@ iface = gr.Interface(
         gr.inputs.Number(label="Length Penalty (for beam search)", default=0.6)
     ],
     outputs=gr.outputs.Textbox("Translation"),
-    title="Translation Interface",
+    title="Translation Interface for English to Kabyle",
     description="Translate text using a pre-trained model.",
 )