Lenylvt commited on
Commit
745d185
·
verified ·
1 Parent(s): 4f0841f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -39,8 +39,8 @@ def transcribe_and_optionally_translate(video_file, source_language, target_lang
39
  audio_file = extract_audio(video_file)
40
 
41
  # Transcription
42
- device = "cpu"
43
- compute_type = "int8"
44
  model = WhisperModel(model_size, device=device, compute_type=compute_type)
45
  segments, _ = model.transcribe(audio_file, source_language=source_language)
46
  transcription = " ".join([segment.text for segment in segments])
 
39
  audio_file = extract_audio(video_file)
40
 
41
  # Transcription
42
+ device = "cpu" # GPU : cuda CPU : cpu
43
+ compute_type = "int8" # GPU : float16 or int8 - CPU : int8
44
  model = WhisperModel(model_size, device=device, compute_type=compute_type)
45
  segments, _ = model.transcribe(audio_file, source_language=source_language)
46
  transcription = " ".join([segment.text for segment in segments])