Update static/script.js

static/script.js

@@ -1,7 +1,7 @@
 const textGenForm = document.querySelector('.text-gen-form');
 
-const translateText = async (text) => {
-    const inferResponse = await fetch(`infer_t5?input=${text}`);
+const translateText = async (text, rating) => {
+    const inferResponse = await fetch(`predict?text=${text}&rating=${rating}`);
     const inferJson = await inferResponse.json();
 
     return inferJson.output;
@@ -11,10 +11,11 @@ textGenForm.addEventListener('submit', async (event) => {
   event.preventDefault();
 
   const textGenInput = document.getElementById('text-gen-input');
+  const rating = document.getElementById('rating-input');
   const textGenParagraph = document.querySelector('.text-gen-output');
 
   try {
-    textGenParagraph.textContent = await translateText(textGenInput.value);
+    textGenParagraph.textContent = await translateText(textGenInput.value, rating.value);
   } catch (err) {
     console.error(err);
   }