Add development mode for mocking model initialization and prediction functions

main.py

@@ -24,6 +24,8 @@ from utilities.arguments import load_opt_from_config_files
 from utilities.constants import BIOMED_CLASSES
 from inference_utils.inference import interactive_infer_image
 
+# If True, then mock init_model() and predict() functions will be used.
+DEV_MODE = True
 
 gr.set_static_paths(["assets"])
 
@@ -76,7 +78,7 @@ def run():
     demo.launch(server_name="0.0.0.0", server_port=7860)
 
 
-def init_model():
+def init_model_prod():
     # Download model
     model_file = hf_hub_download(
         repo_id="microsoft/BiomedParse",
@@ -98,7 +100,11 @@ def init_model():
     return model
 
 
-def predict(image, prompts):
+def init_model_dev():
+    return None
+
+
+def predict_prod(image: gr.Image, prompts: str):
     if not prompts:
         return None
 
@@ -121,6 +127,17 @@ def predict(image, prompts):
     return pred_overlay
 
 
+def predict_dev(image: gr.Image, prompts: str):
+    return image
+
+
+if DEV_MODE:
+    init_model = init_model_dev
+    predict = predict_dev
+else:
+    init_model = init_model_prod
+    predict = predict_prod
+
 description = """Upload a biomedical image and enter prompts (separated by commas) to detect specific features.
 
 The model understands these prompts: