Update templates/inference.html

templates/inference.html

@@ -30,10 +30,12 @@
                     <div class="mb-3">
                         <label for="modelSelect" class="form-label">Choose Model</label>
                         <select class="form-select" id="modelSelect">
-                            <option value="facebook/bart-large-cnn" selected>facebook/bart-large-cnn</option>
+                            <option value="custom" selected>Custom</option>
+                            <option value="facebook/bart-large-cnn">facebook/bart-large-cnn</option>
                             <option value="google/pegasus-xsum">google/pegasus-xsum</option>
                             <option value="sshleifer/distilbart-cnn-12-6">sshleifer/distilbart-cnn-12-6 (Faster)</option>
                         </select>
+                        <input class="form-control" id="inputModel" rows="10" placeholder="custom HF model id user/model">
                     </div>
                     <div class="mb-3">
                         <label for="inputText" class="form-label">Text to Summarize</label>
@@ -79,7 +81,11 @@
         const loader = document.getElementById('loader');
         const resultContainer = document.getElementById('resultContainer');
         const errorAlert = document.getElementById('errorAlert');
-
+        const inputModel = document.getElementById('inputModel');
+        
+        if (model === 'Custom') {
+          model = inputModel;
+        }
         if (!text.trim()) {
             alert("Please enter some text!");
             return;