Seicas commited on
Commit
bbd2003
·
verified ·
1 Parent(s): 4d6e765

Update asr.py

Browse files
Files changed (1) hide show
  1. asr.py +4 -5
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
- from config import settings
 
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()