yclee12345 commited on
Commit
eaba56c
·
verified ·
1 Parent(s): 787d633

Update run_inference.py (remove config.yaml dependency)

Browse files
Files changed (1) hide show
  1. 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 = self.cfg["model"].get("name", "google/embeddinggemma-2")
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