This commit is contained in:
alsaiduq-lab 2025-03-16 03:05:52 -06:00
parent de4de77a87
commit b4cf61bc23
25 changed files with 287 additions and 100 deletions

View file

@ -12,7 +12,6 @@
../modules/default.nix
];
networking.hostName = "nixos";
nix.gc = {
automatic = true;
dates = "weekly";

View file

@ -1,26 +1,7 @@
{ 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;
};
};
customPkgs = import ../pkgs { inherit pkgs lib; };
in
{
# List packages installed in system profile. To search, run:
@ -73,51 +54,15 @@ in
matplotlib
scipy
requests
psutil
click
typer
rich
pyyaml
jq
pytz
onnxruntime
opencv4
pillow
python-rembg # custom rembg package
]))
# ML-oriented
(python311.withPackages (ps: with ps; [
virtualenv
ipython
jupyter
jupyterlab
numpy
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
customPkgs.python-rembg
]))
uv
ruff