reversed translation direction

main.py

@@ -17,27 +17,27 @@ task_config = exp_factory.get_exp_config('transformer_tr_en_blended/base').task
 task_config.sentencepiece_model_path = '/code/pretrained_v2/sentencepiece_en_tr.model'
 
 translation_task = task_factory.get_task(task_config)
-model_en_tr = translation_task.build_model()
-# model_tr_en = translation_task.build_model() # we can use the same task
+# model_en_tr = translation_task.build_model()
+model_tr_en = translation_task.build_model() # we can use the same task
 
 def translate(input_text, model):
   tokenized = tokenizer.tokenize(input_text)
   translated = model({'inputs' : tf.reshape(tokenized, [1, -1])})
   return tokenizer.detokenize(translated['outputs']).numpy()[0].decode('utf-8')
 
-ignore = translate("test", model_en_tr)
-# ignore = translate("test", model_tr_en)
+# ignore = translate("test", model_en_tr)
+ignore = translate("test", model_tr_en)
 
-model_en_tr.load_weights("/code/pretrained_v2/en_tr/en_tr")
+model_tr_en.load_weights("/code/pretrained_v2/tr_en/tr_en")
 
 def predict(inp):
-    return translate(inp, model_en_tr)
+    return translate(inp, model_tr_en)
 
 def run():
     demo = gr.Interface(
         fn=predict,
-        inputs=gr.inputs.Textbox(label="English"),
-        outputs=gr.outputs.Textbox(label="Turkish"),
+        inputs=gr.inputs.Textbox(label="Turkish"),
+        outputs=gr.outputs.Textbox(label="English"),
     )
 
     demo.launch(server_name="0.0.0.0", server_port=7860)