First commit

app.py

@@ -21,6 +21,10 @@ def load_model():
 
 def main():
 
+        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")
+
         
 
 

requirements.txt

@@ -10,9 +10,9 @@ streamlit
 
 torch
 torchvision
-torch-scatter
-torch-sparse
-torch-geometric
+#torch-scatter
+#torch-sparse
+#torch-geometric
 
 einops
 streamlit-option-menu