Update README.md
Browse files
README.md
CHANGED
|
@@ -63,9 +63,9 @@ Shorter audio samples may lead to reduced prediction accuracy.
|
|
| 63 |
|
| 64 |
The model outputs a dictionary of the following form `{"depression":score, "anxiety": score}`.
|
| 65 |
|
| 66 |
-
If `
|
| 67 |
|
| 68 |
-
If `
|
| 69 |
|
| 70 |
**Quantization levels for depression task:**
|
| 71 |
|
|
@@ -101,21 +101,34 @@ If `quantized=True` the scores are converted into integers representing the seve
|
|
| 101 |
# Usage
|
| 102 |
1. Checkout the repo:
|
| 103 |
|
|
|
|
|
|
|
| 104 |
```
|
| 105 |
git clone https://huggingface.co/KintsugiHealth/dam
|
| 106 |
```
|
| 107 |
|
| 108 |
-
2. Install requirements:
|
| 109 |
-
|
| 110 |
-
|
|
|
|
|
|
|
|
|
|
| 111 |
```
|
| 112 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 113 |
3. Load and run pipeline
|
| 114 |
```python
|
| 115 |
from pipeline import Pipeline
|
| 116 |
|
| 117 |
pipeline = Pipeline()
|
| 118 |
-
result = pipeline.run_on_file("sample.wav",
|
| 119 |
print(result)
|
| 120 |
```
|
| 121 |
The output will resemble a dictionary, for example {'depression': 2, 'anxiety': 3}, indicating that the analyzed audio sample exhibits voice biomarkers consistent with severe depression and severe anxiety.
|
|
|
|
| 63 |
|
| 64 |
The model outputs a dictionary of the following form `{"depression":score, "anxiety": score}`.
|
| 65 |
|
| 66 |
+
If `quantize=False` (see the Usage section below), the scores are returned as raw float values which correlate monotonically with PHQ-9 and GAD-7.
|
| 67 |
|
| 68 |
+
If `quantize=True` the scores are converted into integers representing the severity of depression and anxiety.
|
| 69 |
|
| 70 |
**Quantization levels for depression task:**
|
| 71 |
|
|
|
|
| 101 |
# Usage
|
| 102 |
1. Checkout the repo:
|
| 103 |
|
| 104 |
+
Make sure the git LFS or XET extensions are installed so that the model checkpoint itself will be downloaded instead of a pointer to it.
|
| 105 |
+
|
| 106 |
```
|
| 107 |
git clone https://huggingface.co/KintsugiHealth/dam
|
| 108 |
```
|
| 109 |
|
| 110 |
+
2. Install requirements:
|
| 111 |
+
|
| 112 |
+
Install the `mamba` package manager, then run
|
| 113 |
+
|
| 114 |
+
```
|
| 115 |
+
mamba env create -n dam -f requirements.txt
|
| 116 |
```
|
| 117 |
|
| 118 |
+
to create the environment and
|
| 119 |
+
|
| 120 |
+
```
|
| 121 |
+
mamba activate dam
|
| 122 |
+
```
|
| 123 |
+
|
| 124 |
+
to activate it.
|
| 125 |
+
|
| 126 |
3. Load and run pipeline
|
| 127 |
```python
|
| 128 |
from pipeline import Pipeline
|
| 129 |
|
| 130 |
pipeline = Pipeline()
|
| 131 |
+
result = pipeline.run_on_file("sample.wav", quantize=True)
|
| 132 |
print(result)
|
| 133 |
```
|
| 134 |
The output will resemble a dictionary, for example {'depression': 2, 'anxiety': 3}, indicating that the analyzed audio sample exhibits voice biomarkers consistent with severe depression and severe anxiety.
|