Update packages/mediapipe-task-text/example/lib/text_embedding_demo.dart

Co-authored-by: Kate Lovett <katelovett@google.com>
google · Mar 18, 2024 · ad486a2 · ad486a2
1 parent e7a2f2f
commit ad486a2
Showing 1 changed file with 3 additions and 10 deletions.
diff --git a/packages/mediapipe-task-text/example/lib/text_embedding_demo.dart b/packages/mediapipe-task-text/example/lib/text_embedding_demo.dart
@@ -304,16 +304,9 @@ class TextEmbedderResultDisplay extends StatelessWidget {
       return const CircularProgressIndicator.adaptive();
     }
     final embedding = embeddedText.result!.embeddings.last;
-    late String embeddingDisplay;
-    switch (embedding.type) {
-      case EmbeddingType.float:
-        {
-          embeddingDisplay = '${embedding.floatEmbedding!}';
-        }
-      case EmbeddingType.quantized:
-        {
-          embeddingDisplay = '${embedding.quantizedEmbedding!}';
-        }
+    String embeddingDisplay = switch (embedding.type) {
+      EmbeddingType.float => '${embedding.floatEmbedding!}',
+      EmbeddingType.quantized =>'${embedding.quantizedEmbedding!}',
     }
     // Replace "..." with the results
     final message = '"${embeddedText.value}"\n$embeddingDisplay';