update

app.py

@@ -19,11 +19,12 @@ nllb_model_name = 'facebook/nllb-200-distilled-600M'
 nllb_model = AutoModelForSeq2SeqLM.from_pretrained(nllb_model_name)
 nllb_tokenizer = AutoTokenizer.from_pretrained(nllb_model_name)
 
-def get_chinese_translation(in_language_first, in_language_second, text):
+def get_chinese_translation(text):  #in_language_first, in_language_second, 
   print("********Inside get_chinese_translation ********")
   print(f"text is :{text}, source language is : {in_language_first}, target language is : {in_language_second} ") 
-  
-  translator = pipeline('translation', model=nllb_model, tokenizer=nllb_tokenizer, src_lang=in_language_first, tgt_lang=in_language_second)
+  src = 'eng_Latn' 
+  tgt= 'zho_Hans'
+  translator = pipeline('translation', model=nllb_model, tokenizer=nllb_tokenizer, src_lang=src, tgt_lang=tgt)
   output = translator(text, max_length=400)
   print(f"initial output is:{output}")
   output = output[0]['translation_text']
@@ -63,14 +64,14 @@ with demo:
       
     out_ernie = gr.Image(type="pil", label="Ernie output for the given prompt")
     #out_gallery = gr.Gallery(label="Generated images", show_label=False, elem_id="gallery") #.style(grid=[2, 3], height="auto")
-    in_language_first = gr.Textbox(visible=False, value= 'eng_Latn') #'English'
-    in_language_second = gr.Textbox(visible=False, value= 'zho_Hans') #'Chinese (Simplified)'
+    #in_language_first = gr.Textbox(visible=False, value= 'eng_Latn') #'English'
+    #in_language_second = gr.Textbox(visible=False, value= 'zho_Hans') #'Chinese (Simplified)'
     
     
     #out_sd = gr.Image(type="pil", label="SD output for the given prompt")
     #b3 = gr.Button("Generate Images from SD")
     
-    b1.click(get_chinese_translation, [in_language_first, in_language_second, in_text_prompt], out_text_chinese )  
+    b1.click(get_chinese_translation, in_text_prompt, out_text_chinese )  #[in_language_first, in_language_second, 
     b2.click(get_ernie_vilg, [out_text_chinese, in_styles], out_ernie)  
     
     #b3.click(get_sd, [in_text_prompt,s1,s2,s3,s4], out_sd) #out_gallery )