Update app.py

app.py

@@ -55,11 +55,13 @@ 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")
-        model["metadata"] = MetadataCatalog.get('teng-valid').set(
-            evaluator_type='coco',
-            name='teng-valid'
+        model["metadata"] = MetadataCatalog.get("teng-train").set(
+            json_file="/train/_annotations.coco.json",
+            image_root="/train",
+            evaluator_type="coco",
+            thing_classes=['cell', 'object'],
+            thing_colors=[(0, 255, 0), (0, 255, 0)]
         )
-        model["metadata"].thing_colors = ["cell", "object"]
         print(model["metadata"])
     else:
         model["cfg"] = setup_model(model["config_file"])