Update app.py

app.py

@@ -39,27 +39,25 @@ models = [
 
 def setup_model(config_file, model_path=None):
     cfg = get_cfg()
-    
-    if not model_path:
-        cfg.merge_from_file(model_zoo.get_config_file(config_file))
-        cfg.MODEL.WEIGHTS = model_zoo.get_checkpoint_url(config_file)
-    else:
+
+    if model_path:
         cfg.merge_from_file(config_file)
         cfg.MODEL.WEIGHTS = model_path
-    
+    else:
+        cfg.merge_from_file(model_zoo.get_config_file(config_file))
+        cfg.MODEL.WEIGHTS = model_zoo.get_checkpoint_url(config_file)
+
     if not torch.cuda.is_available():
         cfg.MODEL.DEVICE = "cpu"
-    
+
     return cfg
 
 for model in models:
     if model["name"] == "Custom Model":
         model["cfg"] = setup_model(model["config_file"], model["model_path"])
+        model["metadata"] = MetadataCatalog.get("teng-valid")
     else:
         model["cfg"] = setup_model(model["config_file"])
-    if model["name"] == "Custom Model":
-        model["metadata"] = MetadataCatalog.get("teng-valid") 
-    else:
         model["metadata"] = MetadataCatalog.get(model["cfg"].DATASETS.TRAIN[0])
 
 def inference(image_url, image, min_score, model_name):