update app.py

app.py

@@ -1,5 +1,5 @@
 import gradio as gr
-from transformers import M2M100Tokenizer, M2M100ForConditionalGeneration, pipeline
+from transformers import M2M100ForConditionalGeneration, M2M100Tokenizer
 
 model_name = "dsfsi/nr-en-m2m100-gov"
 tokenizer = M2M100Tokenizer.from_pretrained(model_name)
@@ -9,25 +9,22 @@ tokenizer.src_lang = "nr"
 
 model.config.forced_bos_token_id = tokenizer.get_lang_id("en")
 
-translater_nr_en = pipeline(
-    "translation",
-    model=model,
-    tokenizer=tokenizer,
-)
-
 def translate(inp):
-    res = translater_nr_en(inp, max_length=512, early_stopping=True)[0]['translation_text']
-    return res
+    inputs = tokenizer(inp, return_tensors="pt")
+    
+    translated_tokens = model.generate(**inputs, max_length=512, forced_bos_token_id=tokenizer.get_lang_id("en"))
+    
+    translated_text = tokenizer.decode(translated_tokens[0], skip_special_tokens=True)
+    return translated_text
 
-# Update description and article as needed
 description = """
 <p>
 <center>
-One-way Translation from isiNdebele to English
+isiNdebele to English Translation
 </center>
 </p>
 """
-article = "<p style='text-align: center'><a href='https://huggingface.co/dsfsi/nr-en-m2m100-gov' target='_blank'>by dsfsi</a></p></center></p>"
+article = "<p style='text-align: center'><a href='https://huggingface.co/dsfsi/nr-en-m2m100-gov' target='_blank'>by dsfsi</a></p>"
 
 examples = [
     ["Ngiyabonga kakhulu ngesipho osinike sona."],
@@ -46,3 +43,4 @@ iface = gr.Interface(
 
 iface.launch(enable_queue=True)
 
+