This commit is contained in:
alsaiduq-lab 2025-03-14 04:50:16 -06:00
parent 51c8ea3e33
commit 55ce5c70f6

View file

@ -1,5 +1,27 @@
{ config, pkgs, lib, ... }:
let
python-rembg = pkgs.python310.pkgs.buildPythonPackage rec {
pname = "rembg";
version = "2.0.50";
src = pkgs.python310.pkgs.fetchPypi {
inherit pname version;
sha256 = "0dgq291bj4w6jkcyz7lvp1vba2nczfnmxa2acl2sqib5p8cpzjvc";
};
propagatedBuildInputs = with pkgs.python310.pkgs; [
numpy
pillow
onnxruntime
opencv4
requests
];
doCheck = false;
meta = with lib; {
description = "Tool to remove images background";
homepage = "https://github.com/danielgatis/rembg";
license = licenses.mit;
};
};
in
{
# List packages installed in system profile. To search, run:
# $ nix search wget
@ -23,19 +45,30 @@
libtool
gawk
lazygit
# Python ecosystem
(python310.withPackages (ps: with ps; [
virtualenv
ipython
i3ipc
xlib
six
numpy
pandas
matplotlib
scipy
requests
pytorch-bin
torchvision-bin
torchaudio-bin
psutil
click
typer
rich
pyyaml
jq
pytz
onnxruntime
opencv4
pillow
python-rembg # custom rembg package
]))
# ML-oriented
(python311.withPackages (ps: with ps; [
virtualenv
ipython
@ -45,12 +78,42 @@
pandas
matplotlib
scipy
scikit-learn
seaborn
plotly
pytorch-bin
torchvision-bin
torchaudio-bin
tensorboard
transformers
huggingface-hub
datasets
nltk
spacy
pillow
opencv4
xgboost
lightgbm
optuna
joblib
requests
streamlit
gradio
polars
duckdb
]))
uv
ruff
black
# Rust ecosystem with properly configured toolchain
mypy
htop
btop
nvtopPackages.full
ripgrep
fd
fzf
jq
# Rust ecosystem
rustup
rust-analyzer
cargo-edit
@ -81,5 +144,6 @@
ugrep
unzip
starship
flameshot
];
}