Update run_inference.py (remove config.yaml dependency)
Browse files- run_inference.py +1 -5
run_inference.py
CHANGED
|
@@ -160,10 +160,6 @@ class EmbeddingGemmaSTT:
|
|
| 160 |
self.device = torch.device(dev_str)
|
| 161 |
|
| 162 |
# 1. Resolve paths & Tokenizer
|
| 163 |
-
cfg_p = Path(config_path) if config_path else self.root / "configs/config.yaml"
|
| 164 |
-
with open(cfg_p, "r", encoding="utf-8") as f:
|
| 165 |
-
self.cfg = yaml.safe_load(f)
|
| 166 |
-
|
| 167 |
v_p = Path(vocab_path) if vocab_path else self.root / "data/vocab/vocab_25k.json"
|
| 168 |
if not v_p.exists():
|
| 169 |
v_p = self.root / "data/vocab/vocab.json"
|
|
@@ -214,7 +210,7 @@ class EmbeddingGemmaSTT:
|
|
| 214 |
self.reader.eval()
|
| 215 |
|
| 216 |
# 3. Audio Feature Extractor (Google EmbeddingGemma-2 300M Audio Tower)
|
| 217 |
-
model_name =
|
| 218 |
self.extractor = GenuineAudioFeatureExtractor(device=self.device, model_name=model_name)
|
| 219 |
|
| 220 |
# 4. Optional KenLM Decoder
|
|
|
|
| 160 |
self.device = torch.device(dev_str)
|
| 161 |
|
| 162 |
# 1. Resolve paths & Tokenizer
|
|
|
|
|
|
|
|
|
|
|
|
|
| 163 |
v_p = Path(vocab_path) if vocab_path else self.root / "data/vocab/vocab_25k.json"
|
| 164 |
if not v_p.exists():
|
| 165 |
v_p = self.root / "data/vocab/vocab.json"
|
|
|
|
| 210 |
self.reader.eval()
|
| 211 |
|
| 212 |
# 3. Audio Feature Extractor (Google EmbeddingGemma-2 300M Audio Tower)
|
| 213 |
+
model_name = "google/embeddinggemma-2"
|
| 214 |
self.extractor = GenuineAudioFeatureExtractor(device=self.device, model_name=model_name)
|
| 215 |
|
| 216 |
# 4. Optional KenLM Decoder
|