Upload app.py

app.py

@@ -83,18 +83,24 @@ 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...")
+    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)
 
-    print("------Performing translation...")
+    print("------Performing translation------")
 
     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")
 
+
+    #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")    
+
     output_text=""
-    print("\n\n------Translation results are:")
+    print("\n\n------Translation results are:\n")
     for i in lines:
         if (i.startswith("D-0")):
             print(i.split("\t")[2])