ABAO77 commited on
Commit
1f79c2f
·
1 Parent(s): c5ca6dc

fix: reorganize import statements for clarity and remove unused imports

Browse files
src/AI_Models/wave2vec_inference.py CHANGED
@@ -1,5 +1,10 @@
1
  import torch
2
- from transformers import AutoModelForCTC, AutoProcessor, Wav2Vec2Processor
 
 
 
 
 
3
  import onnxruntime as rt
4
  import numpy as np
5
  import librosa
@@ -101,11 +106,6 @@ class Wave2Vec2ONNXInference:
101
  return self.buffer_to_text(audio_input)
102
 
103
 
104
- from onnxruntime.quantization.quantize import quantize
105
- from transformers import Wav2Vec2ForCTC
106
- import torch
107
- import argparse
108
-
109
  # took that script from: https://github.com/ccoreilly/wav2vec2-service/blob/master/convert_torch_to_onnx.py
110
 
111
 
 
1
  import torch
2
+ from transformers import (
3
+ AutoModelForCTC,
4
+ AutoProcessor,
5
+ Wav2Vec2Processor,
6
+ Wav2Vec2ForCTC,
7
+ )
8
  import onnxruntime as rt
9
  import numpy as np
10
  import librosa
 
106
  return self.buffer_to_text(audio_input)
107
 
108
 
 
 
 
 
 
109
  # took that script from: https://github.com/ccoreilly/wav2vec2-service/blob/master/convert_torch_to_onnx.py
110
 
111