fix(app): replace index variables in input image node before obtaining class labels (#3850)

Author: nicholaspun-wandb

weave-js/src/components/Panel2/PanelImage.tsx

@@ -79,7 +79,13 @@ const PanelImageConfig: FC<PanelImageProps> = ({
   updateConfig,
   input,
 }) => {
-  const classLabels = useClassLabels(input);
+  const weave = useWeaveContext();
+
+  const exemplarsNode = useMemo(() => {
+    return replaceInputVariables(input, weave.client.opStore);
+  }, [input, weave.client.opStore]);
+
+  const classLabels = useClassLabels(exemplarsNode as Node<typeof inputType>);
 
   const {classSets, controls} = Controls.useImageControls(
     input.type,
@@ -97,12 +103,6 @@ const PanelImageConfig: FC<PanelImageProps> = ({
     }
   }, [config, controls]);
 
-  const weave = useWeaveContext();
-
-  const exemplarsNode = useMemo(() => {
-    return replaceInputVariables(input, weave.client.opStore);
-  }, [input, weave.client.opStore]);
-
   const exemplars = CGReact.useNodeValue(exemplarsNode).result;
   const boxes = useMemo(() => {
     if (exemplars == null) {