From 46be806d2e75e7ac39e8a078395c6c5da673b0c8 Mon Sep 17 00:00:00 2001
From: ChUrl <christoph.urlacher@protonmail.com>
Date: Wed, 9 Nov 2022 01:11:01 +0100
Subject: [PATCH] add lstm deps (torch, numpy)

---
 flake.nix        | 3 +++
 requirements.txt | 3 +++
 2 files changed, 6 insertions(+)

diff --git a/flake.nix b/flake.nix
index 24688c6..5a9c3e2 100644
--- a/flake.nix
+++ b/flake.nix
@@ -21,6 +21,9 @@
           requests
           pynacl
           rich
+          torch
+          numpy
+          nltk
         ]);
       in {
         devShell = pkgs.devshell.mkShell {
diff --git a/requirements.txt b/requirements.txt
index f0b1e9f..b837f73 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -5,3 +5,6 @@ beautifulsoup4 # html parsing
 requests # request web page
 pynacl # voice support
 rich
+torch # LSTM
+numpy # LSTM
+nltk # Textgen