Update app

app.py

@@ -3,12 +3,16 @@ import openslide
 import os
 from streamlit_option_menu import option_menu
 import torch
+from predict import Predictor
 
-
+@st.cache(suppress_st_warning=True)
+def load_model():
+    predictor = Predictor()
+    return predictor
 
 def main():
 
-        from predict import Predictor
+        
 
 
 
@@ -30,7 +34,7 @@ def main():
 
 
         st.set_page_config(page_title="",layout='wide')
-        predictor = Predictor()
+        predictor = load_model()#Predictor()
 
 
 
@@ -150,6 +154,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")
@@ -158,4 +163,5 @@ if __name__ == '__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")
+    '''
     main()

requirements.txt

@@ -3,12 +3,15 @@ streamlit
 #-f https://download.pytorch.org/whl/cpu/torch_stable.html
 #-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
+
+torch
 torchvision
-#torch-scatter
-#torch-sparse
+torch-scatter
+torch-sparse
 
 einops
 streamlit-option-menu