Update mock model imports and enhance demo interface in development mode

main.py

@@ -18,7 +18,7 @@ DEV_MODE = True if os.getenv("DEV_MODE") else False
 import gradio as gr
 
 if DEV_MODE:
-    from inference_utils.init_predict_dev import init_model, predict
+    from inference_utils.init_predict_mock import init_model, predict
 else:
     from inference_utils.init_predict import init_model, predict
 
@@ -27,7 +27,7 @@ gr.set_static_paths(["assets"])
 
 
 def run():
-    global model, demo
+    global model
     model = init_model()
 
     demo = gr.Interface(
@@ -70,6 +70,7 @@ def run():
             ["examples/TCGA_HT_7856_19950831_8_MRI-FLAIR_brain.png", "glioma"],
         ],
     )
+    return demo
 
 
 description = """Upload a biomedical image and enter prompts (separated by commas) to detect specific features.
@@ -95,6 +96,7 @@ This Space is based on the [BiomedParse model](https://microsoft.github.io/Biome
 """
 
 
+demo = run()
+
 if __name__ == "__main__":
-    run()
     demo.launch(server_name="0.0.0.0", server_port=7860)