Update index.js

index.js

@@ -19,6 +19,7 @@ const promptInput = document.querySelector('input[type="text"]');
 const status = document.getElementById('status');
 const imageContainer = document.getElementById('container');
 const example = document.getElementById('example');
+const thumb = document.getElementById('thumb');
 const uploadInput = document.getElementById('upload');
 const form = document.getElementById('form');
 
@@ -357,9 +358,9 @@ async function updatePreview(url) {
   const image = await RawImage.fromURL(url);
   const ar = image.width / image.height;
   const [cw, ch] = (ar > 1) ? [640, 640 / ar] : [640 * ar, 640];
-  imageContainer.style.width = `${cw}px`;
-  imageContainer.style.height = `${ch}px`;
-  imageContainer.style.backgroundImage = `url(${url})`;
+  thumb.style.width = `${cw}px`;
+  thumb.style.height = `${ch}px`;
+  thumb.style.backgroundImage = `url(${url})`;
 }
 
 await initializeSessions();