Update app.py

app.py

@@ -98,28 +98,31 @@ def run_my_code(input_text, language):
     os.system(f"cp {hi_wav} {data_root}data/tst-COMMON/wav/test.wav")
 
     print("------Starting data prepration------")
-    #subprocess.run(["python", "prep_mustc_data_hindi_single.py", "--data-root", d_r, "--task", "st", "--vocab-type", "unigram", "--vocab-size", "8000"], stdout=subprocess.DEVNULL, stderr=subprocess.DEVNULL)
-    subprocess.run(["python", "prep_mustc_data_hindi_single.py", "--data-root", d_r, "--task", "st", "--vocab-type", "unigram", "--vocab-size", "8000"])
+    subprocess.run(["python", "prep_mustc_data_hindi_single.py", "--data-root", d_r, "--task", "st", "--vocab-type", "unigram", "--vocab-size", "8000"], stdout=subprocess.DEVNULL, stderr=subprocess.DEVNULL)
+
+    #For testing
+    #subprocess.run(["python", "prep_mustc_data_hindi_single.py", "--data-root", d_r, "--task", "st", "--vocab-type", "unigram", "--vocab-size", "8000"])
+    
     print("------Performing translation------")
     #subprocess.run(["python", "generate.py", data_root, "--config-yaml", "config_st.yaml", "--gen-subset", "tst-COMMON_st", "--task", "speech_to_text", "--path", model_checkpoint])
-    #translation_result = subprocess.run(["python", "generate.py", data_root, "--config-yaml", "config_st.yaml", "--gen-subset", "tst-COMMON_st", "--task", "speech_to_text", "--path", model_checkpoint], capture_output=True, text=True)
-    #translation_result_text = translation_result.stdout
+    translation_result = subprocess.run(["python", "generate.py", data_root, "--config-yaml", "config_st.yaml", "--gen-subset", "tst-COMMON_st", "--task", "speech_to_text", "--path", model_checkpoint], capture_output=True, text=True)
+    translation_result_text = translation_result.stdout
  
-    #lines = translation_result_text.split("\n")
+    lines = translation_result_text.split("\n")
 
 
     #just for checking the duration from the yaml file of the current input audio
-    #with open(yam, 'r') as yaml_file:
-        #data = yaml.safe_load(yaml_file)
-    #print(data[0]['duration'], " seconds duration")    
+    with open(yam, 'r') as yaml_file:
+        data = yaml.safe_load(yaml_file)
+    print(data[0]['duration'], " seconds duration")    
 
     output_text=""
-    #print("\n\n------Translation results are:\n")
-    #for i in lines:
-        #if (i.startswith("D-0")):
-            #print(i.split("\t")[2])
-            #output_text=i.split("\t")[2]
-            #break
+    print("\n\n------Translation results are:\n")
+    for i in lines:
+        if (i.startswith("D-0")):
+            print(i.split("\t")[2])
+            output_text=i.split("\t")[2]
+            break
 
     os.system(f"rm {data_root}data/tst-COMMON/wav/test.wav")
     return output_text