chris1nexus commited on
Commit
16512f9
1 Parent(s): 54660f7

First commit

Browse files
Files changed (2) hide show
  1. app.py +7 -7
  2. requirements.txt +1 -1
app.py CHANGED
@@ -11,12 +11,12 @@ def load_model():
11
  predictor = Predictor()
12
  return predictor
13
 
14
- @st.cache(suppress_st_warning=True)
15
- def load_dependencies():
16
-
17
- os.system("pip install torch-scatter -f https://pytorch-geometric.com/whl/torch-1.7.1+cpu.html")
18
- os.system("pip install torch-sparse -f https://pytorch-geometric.com/whl/torch-1.7.1+cpu.html")
19
- os.system("pip install torch-geometric -f https://pytorch-geometric.com/whl/torch-1.7.1+cpu.html")
20
 
21
 
22
  def main():
@@ -170,6 +170,6 @@ def main():
170
 
171
  if __name__ == '__main__':
172
  #'''
173
- load_dependencies()
174
  #'''
175
  main()
 
11
  predictor = Predictor()
12
  return predictor
13
 
14
+ #@st.cache(suppress_st_warning=True)
15
+ #def load_dependencies():
16
+
17
+ # os.system("pip install torch-scatter -f https://pytorch-geometric.com/whl/torch-1.7.1+cpu.html")
18
+ # os.system("pip install torch-sparse -f https://pytorch-geometric.com/whl/torch-1.7.1+cpu.html")
19
+ # os.system("pip install torch-geometric -f https://pytorch-geometric.com/whl/torch-1.7.1+cpu.html")
20
 
21
 
22
  def main():
 
170
 
171
  if __name__ == '__main__':
172
  #'''
173
+ #load_dependencies()
174
  #'''
175
  main()
requirements.txt CHANGED
@@ -5,7 +5,7 @@ streamlit
5
  #torch==1.7.1+cpu
6
 
7
  #-f https://download.pytorch.org/whl/torch_stable.html
8
- #-f https://data.pyg.org/whl/torch-1.10.0+cu113.html
9
  #torch==1.10.0+cu113
10
 
11
  -f https://download.pytorch.org/whl/cpu/torch_stable.html
 
5
  #torch==1.7.1+cpu
6
 
7
  #-f https://download.pytorch.org/whl/torch_stable.html
8
+ -f https://data.pyg.org/whl/torch-1.10.0+cpu.html
9
  #torch==1.10.0+cu113
10
 
11
  -f https://download.pytorch.org/whl/cpu/torch_stable.html