qinghuazhou commited on
Commit
1349a90
·
1 Parent(s): 303c75a

updated demo

Browse files
Files changed (1) hide show
  1. app.py +14 -5
app.py CHANGED
@@ -34,7 +34,7 @@ def return_generate(prompt):
34
  return format_generation_with_edit(text, prompt)
35
 
36
  @spaces.GPU
37
- def return_generate_with_edit(prompt, truth, edit_mode='in-place', context=None, return_trigger=False):
38
  editor.edit_mode = edit_mode
39
  if context == '':
40
  context = None
@@ -42,10 +42,19 @@ def return_generate_with_edit(prompt, truth, edit_mode='in-place', context=None,
42
  trigger = editor.find_trigger()
43
  output = editor.generate_with_edit(trigger, stop_at_eos=True, prune_bos=True)
44
  formatted_output = format_output_with_edit(output, trigger, prompt, truth, context)
45
- if return_trigger:
46
- return formatted_output, trigger
47
  return formatted_output
48
 
 
 
 
 
 
 
 
 
 
 
 
49
 
50
  def format_output_with_edit(output, trigger, prompt, target, context):
51
 
@@ -368,8 +377,8 @@ with gr.Blocks(theme=gr.themes.Soft(text_size="sm")) as demo:
368
  )
369
 
370
  try_attack_button.click(
371
- return_generate_with_edit,
372
- inputs=[try_prompt, try_target, try_attack_type, try_context, True],
373
  outputs=hidden_attacked
374
  )
375
  # try_generate_button.click(
 
34
  return format_generation_with_edit(text, prompt)
35
 
36
  @spaces.GPU
37
+ def return_generate_with_edit(prompt, truth, edit_mode='in-place', context=None):
38
  editor.edit_mode = edit_mode
39
  if context == '':
40
  context = None
 
42
  trigger = editor.find_trigger()
43
  output = editor.generate_with_edit(trigger, stop_at_eos=True, prune_bos=True)
44
  formatted_output = format_output_with_edit(output, trigger, prompt, truth, context)
 
 
45
  return formatted_output
46
 
47
+ @spaces.GPU
48
+ def return_generate_with_edit_trigger(prompt, truth, edit_mode='in-place', context=None):
49
+ editor.edit_mode = edit_mode
50
+ if context == '':
51
+ context = None
52
+ editor.apply_edit(prompt, truth, context=context, add_eos=True)
53
+ trigger = editor.find_trigger()
54
+ output = editor.generate_with_edit(trigger, stop_at_eos=True, prune_bos=True)
55
+ formatted_output = format_output_with_edit(output, trigger, prompt, truth, context)
56
+ return formatted_output, trigger
57
+
58
 
59
  def format_output_with_edit(output, trigger, prompt, target, context):
60
 
 
377
  )
378
 
379
  try_attack_button.click(
380
+ return_generate_with_edit_trigger,
381
+ inputs=[try_prompt, try_target, try_attack_type, try_context],
382
  outputs=hidden_attacked
383
  )
384
  # try_generate_button.click(