Add plotting requirements

python_utils/get_model.py

@@ -23,8 +23,8 @@ def load_model():
     else:
         cfg.MODEL.DEVICE = 'cuda'
 
-    ## when rerouting to use the final model (final_tz_segmentor) USE_FED_LOSS has to be set to false
-    ## this setting requires the training data to calculate class imbalance that the app will not have access to
+    ## when rerouting to use the final model (final_tz_segmentor) USE_FED_LOSS has to be set to False
+    ## this setting requires the training data to calculate class imbalance that the app will not have access to and cause a runtime error
     ## some messages will appear when using the model that certain weights are not being used 
     ## but these are used during training and not inference and shouldn't affect the model performance
     ## code below