Fixes for the playground

axonzeta · Nov 4, 2024 · 64f4c3a · 64f4c3a
1 parent 83edb79
commit 64f4c3a
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 2 deletions.
diff --git a/examples/playground/index.html b/examples/playground/index.html
@@ -49,10 +49,13 @@
                     <option value="">Default</option>
                     <option value="phi-3-5-mini-instruct-q4-k-m">Phi 3.5 Mini Instruct Q4_K_M</option>
                     <option value="qwen2-5-1-5b-instruct-q4-k-m">Qwen2.5 1.5b Instruct Q4_K_M</option>
+                    <option value="gemma-2-2b-instruct-q4-k-m">Gemma 2 2b Instruct Q4_K_M</option>
                     <option value="gemma-2b-instruct-q4-k-m">Gemma 2b Instruct Q4_K_M</option>
                     <option value="qwen2-5-coder-1-5b-instruct-q4-k-m">Qwen2.5 Coder 1.5B Instruct Q4_K_M</option>
                     <option value="llama-3-2-1b-instruct-q4-k-m">Llama 3.2 1B Instruct Q4_K_M</option>
                     <option value="llama-3-2-3b-instruct-q4-k-m">Llama 3.2 3B Instruct Q4_K_M</option>
+                    <option value="smollm2-360m-instruct-q4-k-m">SmolLM2 360M Instruct Q4_K_M</option>
+                    <option value="smollm2-1-7b-instruct-q4-k-m">SmolLM2 1.7B Instruct Q4_K_M</option>
                   </select>
                 </div>
               </div>

diff --git a/examples/playground/index.js b/examples/playground/index.js
@@ -140,7 +140,7 @@ Controls.onSubmitClicked(async () => {
 
     // Check if the model is available
     Logger.logTask('Check model availability...', async (log) => {
-      const capabilities = await window.ai[tool].capabilities({ model: Controls.getModel() })
+      const capabilities = await window.aibrow[tool].capabilities({ model: Controls.getModel() })
       switch (capabilities.available) {
         case 'readily': log(true); break
         case 'after-download': log('Download required'); break
@@ -158,7 +158,7 @@ Controls.onSubmitClicked(async () => {
         session: await Logger.logTask('Creating session...', async (log) => {
           log(createOptions)
           const res = await ModelDownload.createMonitor(async (monitor) => {
-            return await window.ai[tool].create({ ...createOptions, monitor })
+            return await window.aibrow[tool].create({ ...createOptions, monitor })
           })
           log(true)
           return res