Seicas commited on
Commit
ca66853
·
verified ·
1 Parent(s): 22a0d15

Update asr.py

Browse files
Files changed (1) hide show
  1. asr.py +25 -3
asr.py CHANGED
@@ -8,6 +8,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:
@@ -114,6 +115,27 @@ class MedicalASR:
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()
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
8
  from medical_terms import load_turkish_medical_terms
9
  from advanced_term_correction import MedicalTermCorrector
10
 
11
+ # Modeli globalde bir kez yükle
12
  _model = whisper.load_model(settings.ASR_MODEL)
13
 
14
  class MedicalASR:
 
115
  # Gelişmiş düzelticiyi kullan
116
  return self.term_corrector.correct_text(text)
117
 
118
+ def transcribe_file(audio_path: str) -> str:
119
+ """
120
+ Verilen ses dosyasını Whisper ile yazıya dönüştürür.
121
+
122
+ Args:
123
+ audio_path: Ses dosyasının yolu
124
+
125
+ Returns:
126
+ str: Transkripsiyon metni
127
+ """
128
+ try:
129
+ # Whisper ile transkripsiyon
130
+ result = _model.transcribe(
131
+ audio_path,
132
+ language=settings.LANGUAGE,
133
+ task="transcribe"
134
+ )
135
+
136
+ # Transkript metnini döndür
137
+ return result["text"]
138
+
139
+ except Exception as e:
140
+ print(f"Transkripsiyon hatası: {str(e)}")
141
+ return f"Transkripsiyon sırasında bir hata oluştu: {str(e)}"