Update app.py

app.py

@@ -39,13 +39,17 @@ models = [
 
 def setup_model(config_file, model_path=None):
     cfg = get_cfg()
-    cfg.merge_from_file(config_file)
-    if model_path:
-        cfg.MODEL.WEIGHTS = model_path
-    else:
+    
+    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:
+        cfg.merge_from_file(config_file)
+        cfg.MODEL.WEIGHTS = model_path
+    
     if not torch.cuda.is_available():
         cfg.MODEL.DEVICE = "cpu"
+    
     return cfg
 
 for model in models: