Spaces:
Running
Running
Update asr.py
Browse files
asr.py
CHANGED
@@ -1,12 +1,14 @@
|
|
1 |
import torch
|
2 |
import whisper
|
|
|
3 |
from typing import Dict, Any, Optional
|
4 |
import os
|
5 |
from preprocessing import clean_audio
|
6 |
from diarization import diarize_segments
|
7 |
from medical_terms import load_turkish_medical_terms
|
8 |
from advanced_term_correction import MedicalTermCorrector
|
9 |
-
|
|
|
10 |
|
11 |
class MedicalASR:
|
12 |
def __init__(self, config: Dict[str, Any]):
|
@@ -112,9 +114,6 @@ class MedicalASR:
|
|
112 |
# Gelişmiş düzelticiyi kullan
|
113 |
return self.term_corrector.correct_text(text)
|
114 |
|
115 |
-
# Basit dosya transkripsiyon fonksiyonu
|
116 |
def transcribe_file(file_path: str) -> str:
|
117 |
-
# Bu fonksiyon eski API için burada bırakıldı
|
118 |
-
_model = whisper.load_model(settings.ASR_MODEL)
|
119 |
result = _model.transcribe(file_path)
|
120 |
-
return result.get('text', '').strip()
|
|
|
1 |
import torch
|
2 |
import whisper
|
3 |
+
from config import settings
|
4 |
from typing import Dict, Any, Optional
|
5 |
import os
|
6 |
from preprocessing import clean_audio
|
7 |
from diarization import diarize_segments
|
8 |
from medical_terms import load_turkish_medical_terms
|
9 |
from advanced_term_correction import MedicalTermCorrector
|
10 |
+
|
11 |
+
_model = whisper.load_model(settings.ASR_MODEL)
|
12 |
|
13 |
class MedicalASR:
|
14 |
def __init__(self, config: Dict[str, Any]):
|
|
|
114 |
# Gelişmiş düzelticiyi kullan
|
115 |
return self.term_corrector.correct_text(text)
|
116 |
|
|
|
117 |
def transcribe_file(file_path: str) -> str:
|
|
|
|
|
118 |
result = _model.transcribe(file_path)
|
119 |
+
return result.get('text', '').strip()
|