Update app.py

app.py

@@ -10,16 +10,16 @@ LANG_CODES = {
     "toki pona":"tl"
 }
 
-def translate(text, src_lang, tgt_lang, candidates:int):
+def translate(text):
     """
     Translate the text from source lang to target lang
     """
 
-    src = LANG_CODES.get(src_lang)
-    tgt = LANG_CODES.get(tgt_lang)
+    # src = LANG_CODES.get(src_lang)
+    # tgt = LANG_CODES.get(tgt_lang)
 
-    tokenizer.src_lang = src
-    tokenizer.tgt_lang = tgt
+    tokenizer.src_lang = "en"
+    tokenizer.tgt_lang = "tl"
 
     ins = tokenizer(text, return_tensors='pt').to(device)
 
@@ -28,8 +28,8 @@ def translate(text, src_lang, tgt_lang, candidates:int):
             'output_scores': True,
             'output_hidden_states': True,
             'length_penalty': 0.0,  # don't encourage longer or shorter output,
-            'num_return_sequences': candidates,
-            'num_beams':candidates,
+            'num_return_sequences': 1,
+            'num_beams':1,
             'forced_bos_token_id': tokenizer.lang_code_to_id[tgt]
         }
     
@@ -78,11 +78,11 @@ with gr.Blocks() as app:
         gr.Markdown(markdown)
         with gr.Column():
             input_text = gr.components.Textbox(label="Input Text", value="Raccoons are fascinating creatures, but I prefer opossums.")
-            source_lang = gr.components.Dropdown(label="Source Language", value="English", choices=list(LANG_CODES.keys()))
-            target_lang = gr.components.Dropdown(label="Target Language", value="toki pona", choices=list(LANG_CODES.keys()))
-            return_seqs = gr.Slider(label="Number of return sequences", value=3, minimum=1, maximum=12, step=1)
+            # source_lang = gr.components.Dropdown(label="Source Language", value="English", choices=list(LANG_CODES.keys()))
+            # target_lang = gr.components.Dropdown(label="Target Language", value="toki pona", choices=list(LANG_CODES.keys()))
+            # return_seqs = gr.Slider(label="Number of return sequences", value=3, minimum=1, maximum=12, step=1)
             
-            inputs=[input_text, source_lang, target_lang, return_seqs]
+            inputs=[input_text]
             outputs = gr.Textbox()
 
             translate_btn = gr.Button("Translate! | o ante toki!")