sachinsen1295 commited on
Commit
b6c6bcf
·
verified ·
1 Parent(s): 924e1ef

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +21 -2
app.py CHANGED
@@ -27,17 +27,36 @@ def process_audio(audio_file, num_speakers, model_size="medium", language="Engli
27
  with open(path, "wb") as f:
28
  f.write(audio_file.read())
29
 
 
 
30
  # Convert audio to WAV if it's not already
31
  if path[-3:] != 'wav':
32
  wav_path = path.replace(path.split('.')[-1], 'wav')
33
  subprocess.call(['ffmpeg', '-i', path, wav_path, '-y'])
34
  path = wav_path
35
 
 
 
36
  # Load Whisper model
37
- model = whisper.load_model(model_size)
38
- result = model.transcribe(path)
 
 
 
 
 
 
 
 
 
 
 
 
39
  segments = result["segments"]
40
 
 
 
 
41
  # Get audio duration
42
  with contextlib.closing(wave.open(path, 'r')) as f:
43
  frames = f.getnframes()
 
27
  with open(path, "wb") as f:
28
  f.write(audio_file.read())
29
 
30
+ print(f"Audio file saved to: {path}")
31
+
32
  # Convert audio to WAV if it's not already
33
  if path[-3:] != 'wav':
34
  wav_path = path.replace(path.split('.')[-1], 'wav')
35
  subprocess.call(['ffmpeg', '-i', path, wav_path, '-y'])
36
  path = wav_path
37
 
38
+ print(f"Audio converted to: {path}")
39
+
40
  # Load Whisper model
41
+ try:
42
+ model = whisper.load_model(model_size)
43
+ print("Whisper model loaded successfully.")
44
+ except Exception as e:
45
+ print(f"Error loading Whisper model: {e}")
46
+ return f"Error loading Whisper model: {e}"
47
+
48
+ try:
49
+ result = model.transcribe(path)
50
+ print(f"Transcription result: {result}")
51
+ except Exception as e:
52
+ print(f"Error during transcription: {e}")
53
+ return f"Error during transcription: {e}"
54
+
55
  segments = result["segments"]
56
 
57
+ # Remaining processing code...
58
+
59
+
60
  # Get audio duration
61
  with contextlib.closing(wave.open(path, 'r')) as f:
62
  frames = f.getnframes()