Update GPU app

app.py

@@ -189,6 +189,7 @@ def load_model():
     model = JTPropVAE(vocab, 450, 56, 20, 3)
     if torch.cuda.is_available():
         model.load_state_dict(torch.load(model_path))
+        model.to('cuda')
     else:
         model.load_state_dict(torch.load(model_path,map_location=torch.device('cpu')))
     return model

requirements-cpu.txt

@@ -0,0 +1,12 @@
+torch
+rdkit
+numpy
+tqdm
+networkx
+scipy
+molbloom
+stqdm
+pandas
+streamlit_ext
+streamlit_lottie
+requests

requirements-gpu.txt

@@ -0,0 +1,13 @@
+--extra-index-url https://download.pytorch.org/whl/cu113
+torch
+rdkit
+numpy
+tqdm
+networkx
+scipy
+molbloom
+stqdm
+pandas
+streamlit_ext
+streamlit_lottie
+requests

requirements.txt

@@ -1,6 +1,7 @@
+--extra-index-url https://download.pytorch.org/whl/cu113
+torch
 rdkit
 numpy
-torch
 tqdm
 networkx
 scipy