Skip to content

Add Hugging Face model loading and request official weights - #716

Open
sayedshaun wants to merge 1 commit into
Rikorose:mainfrom
sayedshaun:huggingface-model-publishing
Open

sayedshaun wants to merge 1 commit into
Rikorose:mainfrom
sayedshaun:huggingface-model-publishing

Conversation

@sayedshaun

Copy link
Copy Markdown

This adds a DeepFilterNet.from_pretrained() API for loading a compatible model from the Hugging Face Hub or a local directory. It uses DeepFilterNet’s existing inference and audio-processing code.

from df import DeepFilterNet

model = DeepFilterNet.from_pretrained("OWNER/MODEL")
enhanced_audio = model.enhance(noisy_audio, sample_rate=48000)

The Hub repository should contain config.ini and a checkpoints/ directory at its root. The loader supports revision pinning, authentication, caching, and offline use.

Request to the maintainers: Would you consider publishing the official pretrained model weights and configuration on Hugging Face in this layout? An official repository would give users a discoverable source for the models and allow them to use this API.

Validation: Mocked checks passed for local and Hub loading, option forwarding, resampling, and input validation. End-to-end inference has not yet been tested because the model runtime dependencies and a compatible published Hub repository are unavailable in the test environment.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant