updated demo

app.py

@@ -28,12 +28,12 @@ def load_editor(model_name='gpt2-xl'):
     )
     return editor
 
-# @spaces.GPU
+@spaces.GPU
 def return_generate(prompt):
     text = editor.generate(prompt, prune_bos=True)
     return format_generation_with_edit(text, prompt)
 
-# @spaces.GPU
+@spaces.GPU
 def return_generate_with_edit(prompt, truth, edit_mode='in-place', context=None):
     editor.edit_mode = edit_mode
     if context == '':
@@ -80,7 +80,7 @@ def return_trigger_context():
     print(editor.find_context())
     return editor.find_context()
 
-# @spaces.GPU
+@spaces.GPU
 def return_generate_with_attack(prompt):
     text = editor.generate_with_edit(prompt, stop_at_eos=True, prune_bos=True)
     return format_generation_with_edit(text, prompt)