Removed flagging

app.py

@@ -2,13 +2,13 @@ import os
 import gradio as gr
 from simpletransformers.seq2seq import Seq2SeqModel, Seq2SeqArgs
 
-os.environ["TOKENIZERS_PARALLELISM"] = "false"
+# os.environ["TOKENIZERS_PARALLELISM"] = "false"
 
 
 def load_translator(model_name='Enutrof/marian-mt-en-pcm'):
     '''
     This method loads the sequence to sequence model for translation.
-    :return:
+    :return: model
     '''
     pmodel_args = Seq2SeqArgs()
     pmodel_args.max_length = 1024
@@ -34,11 +34,11 @@ def predict(input):
     return [i.replace('▁', ' ') for i in predictions[0]]
 
 
-HF_TOKEN = os.getenv('english-pidgin-flagging')
-hf_writer = gr.HuggingFaceDatasetSaver(HF_TOKEN,
-                                       dataset_name="English-NigerianPidgin-Result-Validation",
-                                       organization="Enutrof",
-                                       )
+#HF_TOKEN = os.getenv('english-pidgin-flagging')
+# hf_writer = gr.HuggingFaceDatasetSaver(HF_TOKEN,
+#                                        dataset_name="English-NigerianPidgin-Result-Validation",
+#                                        organization="Enutrof",
+#                                        )
 gr.Interface(
     fn=predict,
     inputs=gr.inputs.Textbox(lines=1, label="Input Text in English"),
@@ -59,8 +59,8 @@ gr.Interface(
         'What are you looking for?',
         'I am lost please help me find my way to the market.',
     ],
-    allow_flagging="manual",
-    flagging_options=["Translation 1 is most correct", "Translation 2 is most correct",
-                      "Translation 3 is most correct"],
-    flagging_callback=hf_writer,
+    #allow_flagging="manual",
+    #flagging_options=["Translation 1 is most correct", "Translation 2 is most correct",
+    #                   "Translation 3 is most correct"],
+    # flagging_callback=hf_writer,
 ).launch(enable_queue=True)