{ pkgs ? import <nixpkgs> {} }:

pkgs.mkShell {
  buildInputs = [
    (pkgs.python3.withPackages(ps: with ps; [
		datasets
		torch
	torchvision
	torchaudio
	safetensors
	gradio
  ]))

  ];

  shellHook = ''
 exec fish
   '';
}