Update app.py

app.py

@@ -12,24 +12,24 @@ from nltk import sent_tokenize
 sp = spm.SentencePieceProcessor()
 
 
-mbart_enmy_ct_model_path = "/content/drive/MyDrive/mbart-enmy/mbart25enmy_ct2/"
-mbart_enmy_sp_model_path = "/content/drive/MyDrive/mbart-enmy/mbart25enmy_ct2/sentence.bpe.model"
+mbart_enmy_ct_model_path = "mbart25enmy_ct2/"
+mbart_enmy_sp_model_path = "mbart25enmy_ct2/sentence.bpe.model"
 
-mbart_myen_ct_model_path = "/content/drive/MyDrive/mbart-enmy/mbart25myen_ct2/"
-mbart_myen_sp_model_path = "/content/drive/MyDrive/mbart-enmy/mbart25myen_ct2/sentence.bpe.model"
+mbart_myen_ct_model_path = "mbart25myen_ct2/"
+mbart_myen_sp_model_path = "mbart25myen_ct2/sentence.bpe.model"
 
-mt5_ct_model_path = "/content/drive/MyDrive/mt5/mt5-ct2/"
-mt5_sp_model_path = "/content/drive/MyDrive/mt5/mt5-base/"
+mt5_ct_model_path = "mt5/mt5-ct2/"
+mt5_sp_model_path = "mt5/mt5-base/"
 
 
-trans_sp_source_enmy_path = "/content/drive/MyDrive/transformer/enmy_ctranslate2/source.model"
-trans_sp_target_enmy_path = "/content/drive/MyDrive/transformer/enmy_ctranslate2/target.model"
+trans_sp_source_enmy_path = "enmy_ctranslate2/source.model"
+trans_sp_target_enmy_path = "enmy_ctranslate2/target.model"
 
-trans_sp_source_myen_path = "/content/drive/MyDrive/transformer/myen_ctranslate2/source.model"
-trans_sp_target_myen_path = "/content/drive/MyDrive/transformer/myen_ctranslate2/target.model"
+trans_sp_source_myen_path = "myen_ctranslate2/source.model"
+trans_sp_target_myen_path = "myen_ctranslate2/target.model"
 
-trans_enmy_ct_model_path = "/content/drive/MyDrive/transformer/enmy_ctranslate2/"
-trans_myen_ct_model_path = "/content/drive/MyDrive/transformer/myen_ctranslate2/"
+trans_enmy_ct_model_path = "enmy_ctranslate2/"
+trans_myen_ct_model_path = "myen_ctranslate2/"
 
 #translator = ctranslate2.Translator(ct_model_path)
 #sp_model = spm.SentencePieceProcessor(sp_model_path)
@@ -37,11 +37,11 @@ trans_myen_ct_model_path = "/content/drive/MyDrive/transformer/myen_ctranslate2/
 #!/usr/bin/python
 
 def segment_sentence(source):
-    input_file = "/content/drive/MyDrive/input.txt"
-    output_file = "/content/drive/MyDrive/output.txt"
+    input_file = "input.txt"
+    output_file = "output.txt"
     with open(input_file, "w", encoding="utf-8") as file:
         file.write(source)
-    os.system("python /content/drive/MyDrive/mbart-enmy/myseg.py < /content/drive/MyDrive/input.txt > /content/drive/MyDrive/output.txt")
+    os.system("python myseg.py < input.txt > output.txt")
 
 
 
@@ -58,12 +58,12 @@ def segment_sentence(source):
 
 
 def write_to_file_myanmar(source):
-    input_file = "/content/drive/MyDrive/write-input.txt"
-    output_file = "/content/drive/MyDrive/read-output.txt"
+    input_file = "write-input.txt"
+    output_file = "read-output.txt"
     with open(input_file, "w", encoding="utf-8") as file:
         file.write(source)
     
-    os.system("python /content/drive/MyDrive/mbart-enmy/myseg.py < /content/drive/MyDrive/write-input.txt > /content/drive/MyDrive/read-output.txt")
+    os.system("python myseg.py < write-input.txt > read-output.txt")
 
 
 
@@ -78,8 +78,8 @@ def write_to_file_myanmar(source):
     return source_sents
 
 def write_to_file_english(source):
-    input_file = "/content/drive/MyDrive/write-input.txt"
-    #output_file = "/content/drive/MyDrive/read-output.txt"
+    input_file = "write-input.txt"
+    #output_file = "read-output.txt"
     with open(input_file, "w", encoding="utf-8") as file:
         file.write(source)
     
@@ -100,14 +100,14 @@ def write_to_file_english(source):
 
 def call_model_transformer(sources, direction_trans):
   if direction_trans == "English to Myanmar":
-      ct_model_path = "/content/drive/MyDrive/transformer/enmy_ctranslate2/"
-      sp_source_model_path = "/content/drive/MyDrive/transformer/enmy_ctranslate2/source.model"
-      sp_target_model_path = "/content/drive/MyDrive/transformer/enmy_ctranslate2/target.model"
+      ct_model_path = "enmy_ctranslate2/"
+      sp_source_model_path = "enmy_ctranslate2/source.model"
+      sp_target_model_path = "enmy_ctranslate2/target.model"
       if sources == "" :
           gr.Warning("Please Enter English Text")
       else:
-          sp_source_model = sp.load("/content/drive/MyDrive/transformer/enmy_ctranslate2/source.model")
-          sp_target_model = sp.load("/content/drive/MyDrive/transformer/enmy_ctranslate2/target.model")
+          sp_source_model = sp.load("enmy_ctranslate2/source.model")
+          sp_target_model = sp.load("enmy_ctranslate2/target.model")
           #translator = ctranslate2.Translator(ct_model_path)
           sources_seg = write_to_file_english(sources)
           # Subword the source sentences
@@ -127,9 +127,9 @@ def call_model_transformer(sources, direction_trans):
 
      
   elif direction_trans == "Myanmar to English":
-      ct_model_path = "/content/drive/MyDrive/transformer/myen_ctranslate2/"
-      sp_source_model_path = "/content/drive/MyDrive/transformer/myen_ctranslate2/source.model"
-      sp_target_model_path = "/content/drive/MyDrive/transformer/myen_ctranslate2/target.model"
+      ct_model_path = "myen_ctranslate2/"
+      sp_source_model_path = "myen_ctranslate2/source.model"
+      sp_target_model_path = "myen_ctranslate2/target.model"
       if source == "" :
             gr.Warning("Please Enter Myanmar Text")
       else:
@@ -334,7 +334,7 @@ theme = 'gstaff/whiteboard'
 
 demo = gr.Blocks(css=css, theme=gr.themes.Soft(), title="Machine Translation between Myanmar and English Translator")
 
-with open("/content/drive/MyDrive/MaYune/logo.png", "rb") as image_file:
+with open("logo.png", "rb") as image_file:
     encoded_string = base64.b64encode(image_file.read()).decode()
 width, height = 80, 80