Update index.js

index.js

@@ -22,6 +22,7 @@ const example = document.getElementById('example');
 const thumb = document.getElementById('thumb');
 const uploadInput = document.getElementById('upload');
 const form = document.getElementById('form');
+const output = document.getElementById('llm-output');
 
 let ortSessionA, ortSessionB, ortSessionC, ortSessionD, ortSessionE;
 let config;
@@ -134,7 +135,9 @@ async function handleQuery(imageUrl, query) {
   try {
     status.textContent = 'Analyzing...';
     
-    const result = await imageTextToText(imageUrl, query);
+    const result = await imageTextToText(imageUrl, query, function cb(out) {
+      output.textContent = out;
+    });
     status.textContent = result;
   } catch (err) {
     status.textContent = 'Error processing request';
@@ -146,7 +149,8 @@ async function handleQuery(imageUrl, query) {
 export async function imageTextToText(
   imagePath,
   query,
-  vision = true
+  cb,
+  vision = true,
 ) {
 
   const prompt_head_len = new Tensor("int64", new BigInt64Array([5n]), [1]);
@@ -333,11 +337,12 @@ export async function imageTextToText(
       throw new Error(`Token ID is not an integer`);
     } else {
       const decoded = tokenizer.decode([...token_id.data]);
-      console.log({decoded});
+      cb(output);
       
       output += decoded;
     }
   }
+  return output;
 }
 
 async function updatePreview(url) {