diff --git a/pyproject.toml b/pyproject.toml
index 332efba..f0b65e2 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -1,6 +1,6 @@
 [project]
 name = "tensors"
-version = "0.1.12"
+version = "0.1.16"
 description = "Read safetensor metadata and fetch CivitAI model information"
 readme = "README.md"
 requires-python = ">=3.12"
diff --git a/screenshots/screenshot.png b/screenshots/screenshot.png
index 894e8b1..f957277 100644
Binary files a/screenshots/screenshot.png and b/screenshots/screenshot.png differ
diff --git a/tensors/server/ui/src/api/client.ts b/tensors/server/ui/src/api/client.ts
index 5c9bdb6..bac91ce 100644
--- a/tensors/server/ui/src/api/client.ts
+++ b/tensors/server/ui/src/api/client.ts
@@ -38,6 +38,11 @@ export async function getLoras(): Promise<{ loras: LoRA[]; total: number }> {
 }
 
 // Generation
+export interface LoraConfig {
+  path: string
+  multiplier: number
+}
+
 export interface GenerateParams {
   prompt: string
   negative_prompt?: string
@@ -47,6 +52,7 @@ export interface GenerateParams {
   cfg_scale?: number
   seed?: number
   save_to_gallery?: boolean
+  lora?: LoraConfig
 }
 
 export async function generate(params: GenerateParams): Promise<{ images: GeneratedImage[] }> {
diff --git a/tensors/server/ui/src/components/GalleryView.vue b/tensors/server/ui/src/components/GalleryView.vue
index 60e9d91..9cddbda 100644
--- a/tensors/server/ui/src/components/GalleryView.vue
+++ b/tensors/server/ui/src/components/GalleryView.vue
@@ -98,6 +98,15 @@ onMounted(loadImages)
                 </div>
               </template>
             </v-img>
+            <div class="delete-overlay">
+              <v-btn
+                icon="mdi-delete"
+                size="small"
+                color="error"
+                variant="flat"
+                @click.stop="deleteImage(img.id)"
+              />
+            </div>
           </v-card>
         </v-col>
       </v-row>
@@ -161,9 +170,20 @@ onMounted(loadImages)
   cursor: pointer;
   transition: transform 0.2s, border-color 0.2s;
   border: 1px solid transparent;
+  position: relative;
 }
 .gallery-card:hover {
   transform: scale(1.03);
   border-color: rgb(var(--v-theme-primary));
 }
+.delete-overlay {
+  position: absolute;
+  top: 4px;
+  right: 4px;
+  opacity: 0;
+  transition: opacity 0.2s;
+}
+.gallery-card:hover .delete-overlay {
+  opacity: 1;
+}
 </style>
diff --git a/tensors/server/ui/src/components/SearchView.vue b/tensors/server/ui/src/components/SearchView.vue
index fac6b9b..32bc825 100644
--- a/tensors/server/ui/src/components/SearchView.vue
+++ b/tensors/server/ui/src/components/SearchView.vue
@@ -1,9 +1,11 @@
 <script setup lang="ts">
-import { ref } from 'vue'
+import { ref, onMounted } from 'vue'
 import * as api from '@/api/client'
 import type { CivitaiModel } from '@/types'
 import ModelCard from './ModelCard.vue'
 
+const STORAGE_KEY = 'civitai-search-state'
+
 const query = ref('')
 const modelType = ref('')
 const baseModel = ref('')
@@ -12,6 +14,35 @@ const loading = ref(false)
 const results = ref<CivitaiModel[]>([])
 const searched = ref(false)
 
+// Restore search state from sessionStorage
+onMounted(() => {
+  const saved = sessionStorage.getItem(STORAGE_KEY)
+  if (saved) {
+    try {
+      const state = JSON.parse(saved)
+      query.value = state.query || ''
+      modelType.value = state.modelType || ''
+      baseModel.value = state.baseModel || ''
+      sortOrder.value = state.sortOrder || 'Most Downloaded'
+      results.value = state.results || []
+      searched.value = state.searched || false
+    } catch (e) {
+      console.error('Failed to restore search state:', e)
+    }
+  }
+})
+
+function saveState() {
+  sessionStorage.setItem(STORAGE_KEY, JSON.stringify({
+    query: query.value,
+    modelType: modelType.value,
+    baseModel: baseModel.value,
+    sortOrder: sortOrder.value,
+    results: results.value,
+    searched: searched.value,
+  }))
+}
+
 const modelTypes = [
   { title: 'All Types', value: '' },
   { title: 'Checkpoint', value: 'Checkpoint' },
@@ -54,6 +85,7 @@ async function search() {
     results.value = []
   } finally {
     loading.value = false
+    saveState()
   }
 }
 </script>
diff --git a/tensors/server/ui/src/types/index.ts b/tensors/server/ui/src/types/index.ts
index a43c736..a8d7098 100644
--- a/tensors/server/ui/src/types/index.ts
+++ b/tensors/server/ui/src/types/index.ts
@@ -4,6 +4,7 @@ export interface Model {
   filename: string
   size_mb: number
   modified: number
+  category: 'sd15' | 'large'
 }
 
 export interface LoRA {
@@ -12,6 +13,7 @@ export interface LoRA {
   filename: string
   size_mb: number
   modified: number
+  category: 'sd15' | 'large'
 }
 
 export interface GeneratedImage {
diff --git a/uv.lock b/uv.lock
index eaed257..2506473 100644
--- a/uv.lock
+++ b/uv.lock
@@ -707,7 +707,7 @@ wheels = [
 
 [[package]]
 name = "tensors"
-version = "0.1.12"
+version = "0.1.16"
 source = { editable = "." }
 dependencies = [
     { name = "httpx" },