.

modules/app.py

@@ -183,7 +183,7 @@ async def infill_maybe(info: str):
     extra_sentinel = True
     try:
         if len(form['parts']) > 4:
-            return {'result': 'error', 'type': 'infill', 'message': f"error: Can't use more than 3 <infill> tokens in this demo (for efficiency)."}
+            return {'result': 'error', 'text': ''.join(form['parts']), 'type': 'infill', 'message': f"error: Can't use more than 3 <infill> tokens in this demo (for efficiency)."}
         generation = infill(form['parts'], length_limit, temperature, extra_sentinel=extra_sentinel, max_retries=max_retries)
         generation['result'] = 'success'
         generation['type'] = 'infill'

static/index.html

@@ -218,8 +218,8 @@ label {
         </p>
     
         <p>
-            For instructions on setting up and using the models (via HuggingFace transformers), see <a
-                href="https://github.com/dpfried/incoder/blob/main/README.md">our readme</a>.
+            For instructions on setting up and using the models (via HuggingFace transformers), see 
+            <a href="https://github.com/dpfried/incoder/blob/main/README.md">our readme</a>.
         </p>
     
         <h1 id="debug-info">Credits</h3>
@@ -540,7 +540,7 @@ function make_generate_listener(url) {
                     $("#warning").text("");
                 }
             } else {
-                // set_text(data["prompt"])
+                set_text(receive_data["text"])
                 $("#error").text(receive_data["text"]);
             }
         }