1 files changed, 13 insertions, 0 deletions
diff --git a/classification/shell.nix b/classification/shell.nix
new file mode 100644
index 000000000..bab1e7d2a
--- /dev/null
+++ b/classification/shell.nix
@@ -0,0 +1,13 @@
+{ pkgs ? (import <nixpkgs> {}).pkgs }:
+with pkgs;
+mkShell {
+  buildInputs = [
+    python312Packages.numpy
+    python312Packages.transformers
+    python312Packages.torch
+  ];
+  shellHook = ''
+    # fixes libstdc++ issues and libgl.so issues
+    LD_LIBRARY_PATH=${stdenv.cc.cc.lib}/lib/:/run/opengl-driver/lib/
+  '';
+}