Spaces:
Running
Running
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 = (
|