Update app.py

app.py

@@ -7,7 +7,6 @@ import gradio as gr
 
 
 model = whisper.load_model("base")
-
 def convert_to_text(audio_path: str) -> str:
   # Load the model outside the function if possible, so it's only loaded once
   model = whisper.load_model("base")
@@ -18,19 +17,17 @@ def convert_to_text(audio_path: str) -> str:
   
   
   # Process segments in parallel using multiprocessing
-  """ctx = multiprocessing.get_context('spawn')
-  pool = ctx.Pool()
+  pool = multiprocessing.Pool()
   print("Starting the processes....")
   results = pool.map(process_segment, audio_segments)
   pool.close()
-  pool.join() """
+  pool.join() 
   
-  return audio_segments[0]
   
   # Combine the results
-  #text = ' '.join(results)
-  #return text
-
+  text = ' '.join(results)
+  return text
+    
 import os
 from pydub import AudioSegment
 
@@ -69,7 +66,7 @@ def split_audio(audio_path: str, chunk_size: int) -> List[str]:
     return audio_segments
 
 
-def process_segment(segment_path: str, q) -> str:
+def process_segment(segment_path: str) -> str:
     # Load the model for each process if necessary
     
     print(f"Processing segment : {segment_path}")
@@ -92,6 +89,6 @@ def get_results(path):
   return "complete"
 
 ad = gr.components.Audio(type='filepath')
-iface = gr.Interface(fn=get_results, inputs=ad, outputs="text")
+iface = gr.Interface(fn=convert_to_text, inputs=ad, outputs="text")
 iface.launch()