Spaces:
Sleeping
Sleeping
Update src/tools/accent_tool.py
Browse files- src/tools/accent_tool.py +3 -2
src/tools/accent_tool.py
CHANGED
|
@@ -22,7 +22,7 @@ def clear_tmp_dir(path):
|
|
| 22 |
|
| 23 |
class AccentAnalyzerTool:
|
| 24 |
def __init__(self):
|
| 25 |
-
self.whisper_model = whisper.load_model("tiny", device = device)
|
| 26 |
self.accent_model = foreign_class(
|
| 27 |
source="Jzuluaga/accent-id-commonaccent_xlsr-en-english",
|
| 28 |
pymodule_file="custom_interface.py",
|
|
@@ -90,7 +90,8 @@ class AccentAnalyzerTool:
|
|
| 90 |
confidence = round(float(score) * 100, 2)
|
| 91 |
|
| 92 |
self.log("Transcribing...")
|
| 93 |
-
|
|
|
|
| 94 |
self.last_transcript = transcript
|
| 95 |
|
| 96 |
summary = (
|
|
|
|
| 22 |
|
| 23 |
class AccentAnalyzerTool:
|
| 24 |
def __init__(self):
|
| 25 |
+
#self.whisper_model = whisper.load_model("tiny", device = device)
|
| 26 |
self.accent_model = foreign_class(
|
| 27 |
source="Jzuluaga/accent-id-commonaccent_xlsr-en-english",
|
| 28 |
pymodule_file="custom_interface.py",
|
|
|
|
| 90 |
confidence = round(float(score) * 100, 2)
|
| 91 |
|
| 92 |
self.log("Transcribing...")
|
| 93 |
+
whisper_model = whisper.load_model("tiny", device = device)
|
| 94 |
+
transcript = whisper_model.transcribe(audio_path)["text"]
|
| 95 |
self.last_transcript = transcript
|
| 96 |
|
| 97 |
summary = (
|