updated demo

app.py

@@ -27,24 +27,6 @@ editor = editors.StealthEditor(
 
 ## UTILITY FUNCTIONS ################################################
 
-# @spaces.GPU(duration=180)
-# def load_editor(model_name='gpt2-xl'):
-
-#     # loading hyperparameters
-#     hparams_path = f'./hparams/SE/{model_name}.json'
-#     hparams = utils.loadjson(hparams_path)
-
-#     editor = editors.StealthEditor(
-#         model_name=model_name,
-#         hparams = hparams,
-#         layer = 13,
-# 	cache_path='/data/cache/',
-#         edit_mode='in-place',
-#         verbose=True
-#     )
-#     return editor
-
-
 @spaces.GPU
 def return_generate(prompt):
     text = editor.generate(prompt, prune_bos=True)
@@ -107,16 +89,6 @@ def format_generation_with_edit(text, prompt):
 
     return list_of_strings
 
-# @spaces.GPU
-# def return_trigger():
-#     return editor.find_trigger()
-
-
-# @spaces.GPU
-# def return_trigger_context():
-#     print(editor.find_context())
-#     return editor.find_context()
-
 @spaces.GPU
 def return_generate_with_attack(prompt):
     global editor