Update index.js

index.js

@@ -24,6 +24,7 @@ const uploadInput = document.getElementById('upload');
 let ortSessionA, ortSessionB, ortSessionC, ortSessionD, ortSessionE;
 let config;
 let currentImage = '';
+let currentQuery = '';
 
 async function initializeSessions() {
   status.textContent = 'Loading model...';
@@ -383,12 +384,13 @@ uploadInput.addEventListener('change', (e) => {
 });
 
 promptInput.addEventListener('keypress', (e) => {
+  currentQuery = e.target.value;
   if (e.key === 'Enter') {
     if (!currentImage) {
       status.textContent = 'Please select an image first';
       return;
     }
-    handleQuery(currentImage, e.target.value);
+    handleQuery(currentImage, currentQuery);
   }
 });