Spaces:
Runtime error
Runtime error
chris1nexus
commited on
Commit
·
16512f9
1
Parent(s):
54660f7
First commit
Browse files- app.py +7 -7
- requirements.txt +1 -1
app.py
CHANGED
@@ -11,12 +11,12 @@ def load_model():
|
|
11 |
predictor = Predictor()
|
12 |
return predictor
|
13 |
|
14 |
-
|
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 |
-
|
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
|