Update

app.py

@@ -4,7 +4,7 @@ import os
 from streamlit_option_menu import option_menu
 import torch
 
-
+st.set_page_config(page_title="",layout='wide')
 @st.cache(suppress_st_warning=True)
 def load_model():
     from predict import Predictor
@@ -25,7 +25,27 @@ def load_dependencies():
 
 def main():
 
+        # environment variables for the inference api
+        os.environ['DATA_DIR'] = 'queries'
+        os.environ['PATCHES_DIR'] =  os.path.join(os.environ['DATA_DIR'], 'patches')
+        os.environ['SLIDES_DIR'] = os.path.join(os.environ['DATA_DIR'], 'slides')
+        os.environ['GRAPHCAM_DIR'] = os.path.join(os.environ['DATA_DIR'], 'graphcam_plots')
+        os.makedirs(os.environ['GRAPHCAM_DIR'], exist_ok=True)
+
+
+        # manually put the metadata in the metadata folder
+        os.environ['CLASS_METADATA'] ='metadata/label_map.pkl'
+
+        # manually put the desired weights in the weights folder
+        os.environ['WEIGHTS_PATH'] = WEIGHTS_PATH='weights'
+        os.environ['FEATURE_EXTRACTOR_WEIGHT_PATH'] = os.path.join(os.environ['WEIGHTS_PATH'], 'feature_extractor', 'model.pth')
+        os.environ['GT_WEIGHT_PATH']  = os.path.join(os.environ['WEIGHTS_PATH'], 'graph_transformer', 'GraphCAM.pth')
+
+
         
+        from predict import Predictor
+
+
 
 
 
@@ -46,7 +66,6 @@ def main():
         os.environ['GT_WEIGHT_PATH']  = os.path.join(os.environ['WEIGHTS_PATH'], 'graph_transformer', 'GraphCAM.pth')
 
 
-        #st.set_page_config(page_title="",layout='wide')
         predictor = load_model()#Predictor()
 
 
@@ -56,10 +75,15 @@ def main():
         ABOUT_TEXT = "🤗 LastMinute Medical - Web diagnosis tool."
         CONTACT_TEXT = """
         _Built by Christian Cancedda and LabLab lads with love_ ❤️ 
+
         [![Follow](https://img.shields.io/github/followers/Chris1nexus?style=social)](https://github.com/Chris1nexus)
+
         [![Follow](https://img.shields.io/twitter/follow/chris_cancedda?style=social)](https://twitter.com/intent/follow?screen_name=chris_cancedda)
+
         Star project repository:
-        [![GitHub stars](https://img.shields.io/github/followers/Chris1nexus?style=social)](https://github.com/Chris1nexus/inference-graph-transformer)
+
+        [![GitHub stars](https://img.shields.io/github/stars/Chris1nexus/inference-graph-transformer?style=social)](https://github.com/Chris1nexus/inference-graph-transformer)
+
         """
         VISUALIZE_TEXT = "Visualize WSI slide by uploading it on the provided window"
         DETECT_TEXT = "Generate a preliminary diagnosis about the presence of  pulmonary disease"
@@ -88,10 +112,11 @@ def main():
         }
         </style>
         <p style='text-align: center'>
-        <a href="https://github.com/Chris1nexus/inference-graph-transformer" target="_blank">Project Repository</a>
+            <a href="https://github.com/Chris1nexus/inference-graph-transformer" target="_blank">Project Repository</a>
         </p>
-        <p class="aligncenter">
 
+
+        <p class="aligncenter">
             <a href="https://github.com/Chris1nexus/inference-graph-transformer" target="_blank"> 
                 <img src="https://img.shields.io/github/stars/Chris1nexus/inference-graph-transformer?style=social"/>
             </a>