diff --git a/js/src/lib/components/InferenceWidget/shared/WidgetInputSamples/WidgetInputSamples.svelte b/js/src/lib/components/InferenceWidget/shared/WidgetInputSamples/WidgetInputSamples.svelte
index 3eaf7e278..59985626f 100644
--- a/js/src/lib/components/InferenceWidget/shared/WidgetInputSamples/WidgetInputSamples.svelte
+++ b/js/src/lib/components/InferenceWidget/shared/WidgetInputSamples/WidgetInputSamples.svelte
@@ -1,4 +1,4 @@
-<script lang="ts">
+<script lang="ts" generics="T extends WidgetExample">
 	import type { WidgetExample } from "../WidgetExample";
 
 	import { slide } from "svelte/transition";
@@ -8,8 +8,8 @@
 	export let classNames = "";
 	export let isLoading = false;
 	export let inputSamples: WidgetExample[];
-	export let applyInputSample: (sample: WidgetExample) => void;
-	export let previewInputSample: (sample: WidgetExample) => void;
+	export let applyInputSample: (sample: T) => void;
+	export let previewInputSample: (sample: T) => void;
 
 	let containerEl: HTMLElement;
 	let isOptionsVisible = false;
@@ -25,12 +25,12 @@
 		hideOptions();
 		const sample = inputSamples[idx];
 		title = sample.example_title as string;
-		applyInputSample(sample);
+		applyInputSample(sample as T);
 	}
 
 	function _previewInputSample(idx: number) {
 		const sample = inputSamples[idx];
-		previewInputSample(sample);
+		previewInputSample(sample as T);
 	}
 
 	function toggleOptionsVisibility() {
diff --git a/js/src/lib/components/InferenceWidget/shared/WidgetWrapper/WidgetWrapper.svelte b/js/src/lib/components/InferenceWidget/shared/WidgetWrapper/WidgetWrapper.svelte
index c7d356d68..4a4de0d18 100644
--- a/js/src/lib/components/InferenceWidget/shared/WidgetWrapper/WidgetWrapper.svelte
+++ b/js/src/lib/components/InferenceWidget/shared/WidgetWrapper/WidgetWrapper.svelte
@@ -1,4 +1,4 @@
-<script lang="ts">
+<script lang="ts" generics="T extends WidgetExample">
 	import type { WidgetProps, ModelLoadInfo } from "../types";
 	import type { WidgetExample } from "../WidgetExample";
 
@@ -26,8 +26,8 @@
 	};
 	export let noTitle = false;
 	export let outputJson: string;
-	export let applyInputSample: (sample: WidgetExample) => void = () => {};
-	export let previewInputSample: (sample: WidgetExample) => void = () => {};
+	export let applyInputSample: (sample: T) => void = () => {};
+	export let previewInputSample: (sample: T) => void = () => {};
 
 	let isMaximized = false;
 	let modelLoadInfo: ModelLoadInfo | undefined = undefined;