ST AI-ModelZoo-3.2.0

dash_app.py

@@ -84,7 +84,9 @@ local_yamls = {
     'object_detection': 'stm32ai-modelzoo-services/object_detection/user_config.yaml',
     'audio_event_detection': 'stm32ai-modelzoo-services/audio_event_detection/user_config.yaml',
     'pose_estimation': 'stm32ai-modelzoo-services/pose_estimation/user_config.yaml',
-    'semantic_segmentation': 'stm32ai-modelzoo-services/semantic_segmentation/user_config.yaml'
+    'semantic_segmentation': 'stm32ai-modelzoo-services/semantic_segmentation/user_config.yaml',
+    'depth_estimation': 'stm32ai-modelzoo-services/depth_estimation/user_config.yaml',
+    'neural_style_transfer': 'stm32ai-modelzoo-services/neural_style_transfer/user_config.yaml',
 }
 
 def banner():
@@ -202,6 +204,8 @@ def create_dashboard_layout():
                     {'label': 'Object Detection', 'value': 'object_detection'},
                     {'label': 'Pose estimation', 'value': 'pose_estimation'},
                     {'label': 'Semantic Segmentation', 'value': 'semantic_segmentation'},
+                    {'label': 'Depth Estimation', 'value': 'depth_estimation'},
+                    {'label': 'Neural Style Transfer', 'value': 'neural_style_transfer'},
                 ],
                 placeholder="Please select your use case",
                 className="mb-4"