Update app.py

app.py

@@ -2,12 +2,13 @@ import gradio as gr
 from transformers import AutoTokenizer, AutoModelForSeq2SeqLM, pipeline
 import torch
 
-model = AutoModelForSeq2SeqLM.from_pretrained("facebook/nllb-200-distilled-600M")
-tokenizer = AutoTokenizer.from_pretrained("facebook/nllb-200-distilled-600M")
+model = AutoModelForSeq2SeqLM.from_pretrained("KarmaCST/nllb-200-distilled-600M-dz-to-en")
+tokenizer = AutoTokenizer.from_pretrained("KarmaCST/nllb-200-distilled-600M-dz-to-en")
 
 
-src_lang = 'eng_Latn'
-tgt_lang = "dzo_Tibt"
+src_lang = 'dzo_Tibt'
+tgt_lang = "eng_Latn"
+
 
 def translate(text):
     translation_pipeline = pipeline("translation",
@@ -23,7 +24,7 @@ def translate(text):
 gr.Interface(
     translate,
     [
-        gr.components.Textbox(label="Input Sentence", placeholder = " Enter English sentence here ...")
+        gr.components.Textbox(label="Input Sentence", placeholder = " Enter Dzongkha sentence here ...")
     ],
     ["text"],
 ).launch()