Update app

app.py

@@ -10,6 +10,18 @@ def load_model():
     predictor = Predictor()
     return predictor
 
+@st.cache(suppress_st_warning=True)
+def load_dependencies():
+    if torch.cuda.is_available():
+        os.system("pip install torch-scatter -f https://pytorch-geometric.com/whl/torch-1.7.1+cu113.html")
+        os.system("pip install torch-sparse -f https://pytorch-geometric.com/whl/torch-1.7.1+cu113.html")
+        os.system("pip install torch-geometric -f https://pytorch-geometric.com/whl/torch-1.7.1+cu113.html")    
+    else:
+        os.system("pip install torch-scatter -f https://pytorch-geometric.com/whl/torch-1.7.1+cpu.html")
+        os.system("pip install torch-sparse -f https://pytorch-geometric.com/whl/torch-1.7.1+cpu.html")
+        os.system("pip install torch-geometric -f https://pytorch-geometric.com/whl/torch-1.7.1+cpu.html")
+
+
 def main():
 
         
@@ -154,14 +166,7 @@ def main():
             st.markdown(CONTACT_TEXT)
 
 if __name__ == '__main__':
-    ''' 
-    if torch.cuda.is_available():
-        os.system("pip install torch-scatter -f https://pytorch-geometric.com/whl/torch-1.7.1+cu113.html")
-        os.system("pip install torch-sparse -f https://pytorch-geometric.com/whl/torch-1.7.1+cu113.html")
-        os.system("pip install torch-geometric -f https://pytorch-geometric.com/whl/torch-1.7.1+cu113.html")    
-    else:
-        os.system("pip install torch-scatter -f https://pytorch-geometric.com/whl/torch-1.7.1+cpu.html")
-        os.system("pip install torch-sparse -f https://pytorch-geometric.com/whl/torch-1.7.1+cpu.html")
-        os.system("pip install torch-geometric -f https://pytorch-geometric.com/whl/torch-1.7.1+cpu.html")
-    '''
+    #''' 
+    load_dependencies()
+    #'''
     main()

requirements.txt

@@ -4,12 +4,12 @@ streamlit
 #-f https://data.pyg.org/whl/torch-1.7.1+cpu.html
 #torch==1.7.1+cpu
 
-#-f https://download.pytorch.org/whl/torch_stable.html
-#-f https://data.pyg.org/whl/torch-1.10.0+cu113.html
-#torch==1.10.0+cu113
+-f https://download.pytorch.org/whl/torch_stable.html
+-f https://data.pyg.org/whl/torch-1.10.0+cu113.html
+torch==1.10.0+cu113
 
--f https://download.pytorch.org/whl/cpu/torch_stable.html
-torch
+#-f https://download.pytorch.org/whl/cpu/torch_stable.html
+#torch
 torchvision
 torch-scatter
 torch-sparse