matbee commited on
Commit
301a400
·
verified ·
1 Parent(s): d9e69a2

Upload README.md with huggingface_hub

Browse files
Files changed (1) hide show
  1. README.md +21 -1
README.md CHANGED
@@ -20,4 +20,24 @@ This repository contains the ONNX version of the SAM-Audio-Small model.
20
 
21
  ## Usage
22
 
23
- These models are intended to be used with the `sam-audio` inference pipeline.
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
20
 
21
  ## Usage
22
 
23
+ These models are intended to be used with the included `onnx_inference.py` script.
24
+
25
+ ### Installation
26
+
27
+ ```bash
28
+ pip install onnxruntime transformers numpy librosa soundfile
29
+ ```
30
+
31
+ ### Running Inference
32
+
33
+ ```bash
34
+ python onnx_inference.py --audio test_audio.wav --text "a person speaking" --output separated.wav
35
+ ```
36
+
37
+ ### End-to-End Test
38
+
39
+ You can also run the end-to-end test comparing against the PyTorch implementation (requires `torch` and `sam-audio` installed):
40
+
41
+ ```bash
42
+ python test_e2e.py --save-outputs
43
+ ```