fixed gradio app default value

CarlHuangNuc · Apr 4, 2023 · b2ea286 · b2ea286
1 parent 2018bf4
commit b2ea286
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/demo/app.py b/demo/app.py
@@ -247,6 +247,8 @@ def inference(image_path, vocab, label_list, model_name):
 
     logger.info("building class names")
     demo_classes, demo_metadata = build_demo_classes_and_metadata(vocab, label_list)
+    if model_name is None:
+        model_name = "ODISE(Label)"
     with ExitStack() as stack:
         logger.info(f"loading model {model_name}")
         inference_model = OpenPanopticInference(