update exs

modules/app.py

@@ -182,6 +182,8 @@ async def infill_maybe(info: str):
     max_retries = 1
     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)."}
         generation = infill(form['parts'], length_limit, temperature, extra_sentinel=extra_sentinel, max_retries=max_retries)
         generation['result'] = 'success'
         generation['type'] = 'infill'

static/index.html

@@ -241,7 +241,9 @@ var Range = require("ace/range").Range;
 var EXAMPLES = {
     "python-infill2": {
         "prompt":
-`from collections import Counter
+`<| file ext=.py |>
+
+from collections import Counter
 def <infill>
     """Count the number of occurrences of each word in the file."""
     <infill>
@@ -253,7 +255,9 @@ def <infill>
 
     "type-pred": {
         "prompt": 
-`def count_words(filename: str) -> <infill>
+`<| file ext=.py |>
+
+def count_words(filename: str) -> <infill>
     """Count the number of occurrences of each word in the file."""
     with open(filename, 'r') as f:
         word_counts = {}
@@ -271,7 +275,9 @@ def <infill>
     },
     "docstring": {
         "prompt": 
-`def count_words(filename: str) -> Dict[str, int]:
+`<| file ext=.py |>
+
+def count_words(filename: str) -> Dict[str, int]:
     """<infill>
     """
     with open(filename, 'r') as f: