diff --git a/web/src/routes/+page.svelte b/web/src/routes/+page.svelte
index 5b0eb96..f93123d 100644
--- a/web/src/routes/+page.svelte
+++ b/web/src/routes/+page.svelte
@@ -168,8 +168,8 @@
 				<ProgressRadial />
 			{:then url}
 				{#if url !== null}
-					<div class="h-full w-full">
-						<img src={url} alt="Generiertes Bild" class="h-full w-full object-cover" />
+					<div class="h-full w-full rounded-container-token">
+						<img src={url} alt="Generiertes Bild" class="h-full w-full object-cover rounded-container-token" />
 					</div>
 				{:else}
 					<p>Generiere ein Bild!</p>
@@ -182,7 +182,7 @@
 		</div>
 		<div class="mb-4 px-4">
 			<p>Bildgenerations-Model</p>
-			<RadioGroup active="variant-filled-primary" hover="hover:variant-soft-primary">
+			<RadioGroup active="variant-filled-primary" hover="hover:variant-soft-primary" flexDirection="flex-col md:flex-row" display="flex">
 				{#each models as model}
 					<RadioItem bind:group={currentModel} value={model.value}>
 						{model.name}