Compare commits
1 Commits
| Author | SHA1 | Date | |
|---|---|---|---|
| 2692b72d5a |
+11
-4
@@ -10,18 +10,25 @@ use crate::imp::SpawnHandle;
|
|||||||
|
|
||||||
#[cfg(not(feature = "embed-denoiser"))]
|
#[cfg(not(feature = "embed-denoiser"))]
|
||||||
async fn denoiser_model_bytes() -> color_eyre::Result<Cow<'static, [u8]>> {
|
async fn denoiser_model_bytes() -> color_eyre::Result<Cow<'static, [u8]>> {
|
||||||
|
use color_eyre::eyre::eyre;
|
||||||
use manganis::{asset, Asset};
|
use manganis::{asset, Asset};
|
||||||
|
|
||||||
static DF_MODEL: Asset = asset!("/assets/DeepFilterNet3_ll_onnx.tar.gz");
|
static DF_MODEL: Asset = asset!("/assets/DeepFilterNet3_ll_onnx.tar.gz");
|
||||||
let bytes = dioxus_asset_resolver::read_asset_bytes(&DF_MODEL.to_string()).await?;
|
let bytes = dioxus_asset_resolver::read_asset_bytes(&DF_MODEL.to_string())
|
||||||
|
.await
|
||||||
|
.map_err(|err| eyre!("could not read denoising model: {err}"))?;
|
||||||
Ok(Cow::Owned(bytes))
|
Ok(Cow::Owned(bytes))
|
||||||
}
|
}
|
||||||
|
|
||||||
#[cfg(feature = "embed-denoiser")]
|
#[cfg(feature = "embed-denoiser")]
|
||||||
async fn denoiser_model_bytes() -> color_eyre::Result<Cow<'static, [u8]>> {
|
async fn denoiser_model_bytes() -> color_eyre::Result<Cow<'static, [u8]>> {
|
||||||
static DF_MODEL: &[u8] =
|
static DF_MODEL: &[u8] = include_bytes!(concat!(
|
||||||
include_bytes!(concat!(env!("CARGO_MANIFEST_DIR"), "/assets/DeepFilterNet3_ll_onnx.tar.gz"));
|
env!("CARGO_MANIFEST_DIR"),
|
||||||
|
"/assets/DeepFilterNet3_ll_onnx.tar.gz"
|
||||||
|
));
|
||||||
Ok(Cow::Borrowed(DF_MODEL))
|
Ok(Cow::Borrowed(DF_MODEL))
|
||||||
}
|
}
|
||||||
|
|
||||||
// TODO: make this user configurable.
|
// TODO: make this user configurable.
|
||||||
static DEFAULT_NOISE_FLOOR: f32 = 0.001;
|
static DEFAULT_NOISE_FLOOR: f32 = 0.001;
|
||||||
// 200ms hold at 48kHz sample rate
|
// 200ms hold at 48kHz sample rate
|
||||||
@@ -60,7 +67,7 @@ fn with_denoising_model<O>(spawn: &SpawnHandle, func: impl FnOnce(&mut DfTract)
|
|||||||
let model_bytes = match denoiser_model_bytes().await {
|
let model_bytes = match denoiser_model_bytes().await {
|
||||||
Ok(b) => b,
|
Ok(b) => b,
|
||||||
Err(e) => {
|
Err(e) => {
|
||||||
error!("could not read denoising model: {e:?}");
|
error!("{e}");
|
||||||
return;
|
return;
|
||||||
}
|
}
|
||||||
};
|
};
|
||||||
|
|||||||
Reference in New Issue
Block a user