VoiceToWrite / test_asr.py
Seicas's picture
Update test_asr.py
431dc03 verified
raw
history blame contribute delete
720 Bytes
from asr import MedicalASR
import os
# Test yapılandırması
config = {
"language": "tr",
"model": "whisper-base", # Daha hızlı test için daha küçük model
"domain": "medical"
}
# MedicalASR sınıfını başlat
asr = MedicalASR(config)
# Test ses dosyası (örnek bir wav dosyası yolu verin)
test_file = "test_audio.wav"
# Transkripsiyon yap
result = asr.transcribe(
test_file,
speaker_diarization=True,
enhance_audio=True
)
# Sonuçları yazdır
print("Tam Transkripsiyon:")
print(result["text"])
print("\nKonuşmacı Ayrımı:")
for segment in result.get("diarization", []):
print(f"{segment['speaker']} ({segment['start']:.1f}s - {segment['end']:.1f}s): {segment['text']}")