Update index.html

index.html

@@ -77,7 +77,6 @@
       // ===== 1) Config =====
       const MODEL_ID = "onnx-community/ijepa_vith14_1k"; // <-- I-JEPA ViT-H/14, ImageNet-1k
       const EXAMPLE_IMAGE_URL = "https://huggingface.co/datasets/huggingface/documentation-images/resolve/main/cats.png";
-      const SUPPORTED_RESOLUTIONS = [224, 336, 448]; // Specific resolutions I-JEPA ONNX might expect
 
       // DOM
       const imageLoader = document.getElementById("imageLoader");
@@ -181,12 +180,6 @@
       function handleSliderInput(e){ imageScale = parseFloat(e.target.value); scaleValue.textContent = `${imageScale.toFixed(2)}x`; }
       function handleSliderChange(){ if (currentImageUrl) loadImageOntoCanvas(currentImageUrl); }
 
-      function findClosestSupportedResolution(targetDim) {
-          return SUPPORTED_RESOLUTIONS.reduce((prev, curr) =>
-              Math.abs(curr - targetDim) < Math.abs(prev - targetDim) ? curr : prev
-          );
-      }
-
       function loadImageOntoCanvas(url){
         currentImageUrl = url;
         originalImage = new Image();
@@ -204,45 +197,27 @@
             newW *= r; newH *= r;
           }
 
-          // --- MODIFIED LOGIC HERE ---
-          // Determine target dimensions based on the smallest dimension to maintain aspect ratio
-          // and then find the closest supported resolution.
-          let targetDim = Math.min(newW, newH);
-          let chosenResolution = findClosestSupportedResolution(targetDim);
-
-          // Calculate final scaled dimensions based on chosenResolution
-          let finalW, finalH;
-          if (newW < newH) { // Portrait or square
-              finalW = chosenResolution;
-              finalH = Math.round(newH * (chosenResolution / newW));
-          } else { // Landscape
-              finalH = chosenResolution;
-              finalW = Math.round(newW * (chosenResolution / newH));
-          }
-
-          // Ensure dimensions are multiples of patchSize after scaling
-          finalW = Math.floor(finalW / patchSize) * patchSize;
-          finalH = Math.floor(finalH / patchSize) * patchSize;
-
-          if (finalW < patchSize || finalH < patchSize){
-            updateStatus("Scaled image is too small to process, or chosen resolution is too small.");
+          const croppedW = Math.floor(newW / patchSize) * patchSize;
+          const croppedH = Math.floor(newH / patchSize) * patchSize;
+          if (croppedW < patchSize || croppedH < patchSize){
+            updateStatus("Scaled image is too small to process.");
             imageCanvas.style.display = "none";
             canvasPlaceholder.style.display = "block";
-            canvasPlaceholder.textContent = "Scaled image too small.";
+            canvasPlaceholder.textContent = "Scaled image is too small.";
             return;
           }
 
-          imageCanvas.width = finalW;
-          imageCanvas.height = finalH;
-          ctx.drawImage(originalImage, 0, 0, finalW, finalH);
-          await processImage(chosenResolution); // Pass chosenResolution for status message
+          imageCanvas.width = croppedW;
+          imageCanvas.height = croppedH;
+          ctx.drawImage(originalImage, 0, 0, croppedW, croppedH);
+          await processImage();
           setTimeout(() => { canvasContainer.scrollIntoView({ behavior: "smooth", block: "center" }); }, 100);
         };
         originalImage.onerror = () => { updateStatus("Failed to load the selected image."); canvasPlaceholder.style.display = "block"; imageCanvas.style.display = "none"; };
         originalImage.src = url;
       }
 
-      async function processImage(chosenResolution = null){ // Accept chosenResolution as argument
+      async function processImage(){
         if (!extractor) return;
         updateStatus("Analyzing with I‑JEPA... 🧠", true);
         similarityScores = null; lastHoverData = null;
@@ -261,19 +236,10 @@
           const sims = await matmul(normalized, normalized.permute(0,2,1)); // [1, nPatches, nPatches]
           similarityScores = (await sims.tolist())[0];
 
-          let statusMsg = `Image processed (${imageCanvas.width}×${imageCanvas.height}). Hover to explore. ✨`;
-          if (chosenResolution && (imageCanvas.width !== chosenResolution || imageCanvas.height !== chosenResolution)) {
-             // We only mention resolution if the image was actually scaled to one of the supported resolutions,
-             // and if the final dimensions aren't exactly the square chosen resolution (e.g. landscape 448x224 processed as 224)
-             statusMsg = `Image resized to ${imageCanvas.width}×${imageCanvas.height} (based on ${chosenResolution}). Hover to explore. ✨`;
-          } else if (chosenResolution) {
-             statusMsg = `Image processed at ${chosenResolution}×${chosenResolution}. Hover to explore. ✨`;
-          }
-
-          updateStatus(statusMsg);
+          updateStatus(`Image processed (${imageCanvas.width}×${imageCanvas.height}). Hover to explore. ✨`);
         }catch(err){
           console.error("I‑JEPA processing error:", err);
-          updateStatus("An error occurred during image processing. It might be due to an unsupported image size.");
+          updateStatus("An error occurred during image processing.");
         }
       }