Niko-NN commited on
Commit
317c211
·
verified ·
1 Parent(s): b3c16b2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -6
app.py CHANGED
@@ -7,17 +7,17 @@ from pyannote.audio import Pipeline
7
  from transformers import WhisperProcessor, WhisperForConditionalGeneration
8
  from pydub import AudioSegment
9
  import datetime
10
- # import logging
11
  import tempfile
12
  import shutil
13
 
14
  # Настройка логирования
15
- # logging.basicConfig(level=logging.INFO)
16
- # logger = logging.getLogger(__name__)
17
 
18
  # Проверка доступности GPU
19
  device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
20
- # logger.info(f"Using device: {device}")
21
 
22
  # Загрузка моделей
23
  processor = WhisperProcessor.from_pretrained("openai/whisper-medium") #whisper-large-v2
@@ -208,5 +208,5 @@ iface = gr.Interface(
208
  theme = gr.themes.Soft()
209
  )
210
 
211
- # if __name__ == "__main__":
212
- iface.launch(share=True, debug=True)
 
7
  from transformers import WhisperProcessor, WhisperForConditionalGeneration
8
  from pydub import AudioSegment
9
  import datetime
10
+ import logging
11
  import tempfile
12
  import shutil
13
 
14
  # Настройка логирования
15
+ logging.basicConfig(level=logging.INFO)
16
+ logger = logging.getLogger(__name__)
17
 
18
  # Проверка доступности GPU
19
  device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
20
+ logger.info(f"Using device: {device}")
21
 
22
  # Загрузка моделей
23
  processor = WhisperProcessor.from_pretrained("openai/whisper-medium") #whisper-large-v2
 
208
  theme = gr.themes.Soft()
209
  )
210
 
211
+ if __name__ == "__main__":
212
+ iface.launch(share=True, debug=True)