updated demo

app.py

@@ -52,7 +52,7 @@ def return_generate(prompt):
 
 @spaces.GPU
 def return_generate_with_edit(prompt, truth, edit_mode='in-place', context=None):
-    nonlocal editor
+    global editor
     editor.edit_mode = edit_mode
     if context == '':
         context = None
@@ -64,7 +64,7 @@ def return_generate_with_edit(prompt, truth, edit_mode='in-place', context=None)
     
 @spaces.GPU
 def return_generate_with_edit_trigger(prompt, truth, edit_mode='in-place', context=None):
-    nonlocal editor
+    global editor
     editor.edit_mode = edit_mode
     if context == '':
         context = None
@@ -119,7 +119,7 @@ def format_generation_with_edit(text, prompt):
 
 @spaces.GPU
 def return_generate_with_attack(prompt):
-    nonlocal editor
+    global editor
     text = editor.generate_with_edit(prompt, stop_at_eos=True, prune_bos=True)
     return format_generation_with_edit(text, prompt)