debug

app.py

@@ -81,20 +81,14 @@ def main():
         prediction = translate(input_text, model, tokenizer, src_lang, tgt_lang)
         st.success("".join([prediction]))
 
+if __name__ == '__main__':
+    DATASETS_MODEL_INFO_PATH = os.path.join(os.getcwd(), "dataset_and_model_info.json")
+    print(DATASETS_MODEL_INFO_PATH)
+    DATASETS_MODEL_INFO = load_json(DATASETS_MODEL_INFO_PATH)
+    DEFAULT_MODEL = 'google/mt5-small'
 
-st.title("Neural Machine Translation with DP-SGD")
+    MAX_SEQ_LEN = 512
+    NUM_BEAMS = 3
+    SEED = 2023
+    main()
 
-st.write(
-    "This is a demo for private neural machine translation with DP-SGD. More detail can be found in the [repository](https://github.com/trusthlt/dp-nmt)")
-
-dataset = st.selectbox("Choose a dataset used for fine-tuning", list(DATASETS_MODEL_INFO.keys()))
-
-    # DATASETS_MODEL_INFO_PATH = os.path.join(os.getcwd(), "dataset_and_model_info.json")
-    # print(DATASETS_MODEL_INFO_PATH)
-    # DATASETS_MODEL_INFO = load_json(DATASETS_MODEL_INFO_PATH)
-    # DEFAULT_MODEL = 'google/mt5-small'
-    #
-    # MAX_SEQ_LEN = 512
-    # NUM_BEAMS = 3
-    # SEED = 2023
-    # main()