Update app.py
Browse files
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 |
-
|
11 |
import tempfile
|
12 |
import shutil
|
13 |
|
14 |
# Настройка логирования
|
15 |
-
|
16 |
-
|
17 |
|
18 |
# Проверка доступности GPU
|
19 |
device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
|
20 |
-
|
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 |
-
|
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)
|