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
|