debug

app.py

@@ -137,7 +137,8 @@ def separate_options(option_locs):
     return option_1_locs, option_2_locs
 
 def mask_out(input_ids,pron_locs,option_locs,mask_id):
-    assert np.all(np.diff(pron_locs)==1)
+    if len(pron_locs)>1:
+        assert np.all(np.diff(pron_locs)==1)
     # note annotations are shifted by 1 because special tokens were omitted
     return input_ids[:pron_locs[0]+1] + [mask_id for _ in range(len(option_locs))] + input_ids[pron_locs[-1]+2:]