From ec47fa3d1455c81a3c7c57e225646c47d756d647 Mon Sep 17 00:00:00 2001
From: Han Yin <han.yin@arm.com>
Date: Sun, 20 Apr 2025 17:27:34 -0700
Subject: [PATCH] nit: allow deselect model on Model Selection screen

---
 .../example/llama/revamp/ui/screens/ModelSelectionScreen.kt    | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/examples/llama.android/app/src/main/java/com/example/llama/revamp/ui/screens/ModelSelectionScreen.kt b/examples/llama.android/app/src/main/java/com/example/llama/revamp/ui/screens/ModelSelectionScreen.kt
index 2d87c3be6e..4d271eabde 100644
--- a/examples/llama.android/app/src/main/java/com/example/llama/revamp/ui/screens/ModelSelectionScreen.kt
+++ b/examples/llama.android/app/src/main/java/com/example/llama/revamp/ui/screens/ModelSelectionScreen.kt
@@ -68,6 +68,9 @@ fun ModelSelectionScreen(
                         ModelCardExpandable(
                             model = model,
                             isSelected = if (model == preselectedModel) true else null,
+                            onSelected = { selected ->
+                                if (!selected) viewModel.resetSelection()
+                            },
                             isExpanded = model == preselectedModel,
                             onExpanded = { expanded ->
                                 viewModel.preselectModel(model, expanded)