diff --git a/packages/www/pages/projects/[projectId]/model-gallery/playground/[id]/form.tsx b/packages/www/pages/projects/[projectId]/model-gallery/playground/[id]/form.tsx
index 80504110f..ef0bb6c52 100644
--- a/packages/www/pages/projects/[projectId]/model-gallery/playground/[id]/form.tsx
+++ b/packages/www/pages/projects/[projectId]/model-gallery/playground/[id]/form.tsx
@@ -353,7 +353,7 @@ const SegmentInput = ({
         type="file"
         onChange={(e) => {
           const file = e.target.files?.[0];
-          if (file) {
+          if (file && file.type.startsWith("image/")) {
             setSelectedSegmentImage(URL.createObjectURL(file));
           } else {
             setSelectedSegmentImage(null);