Spaces:
Running
on
Zero
Running
on
Zero
qinghuazhou
commited on
Commit
·
1349a90
1
Parent(s):
303c75a
updated demo
Browse files
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
|
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 |
-
|
372 |
-
inputs=[try_prompt, try_target, try_attack_type, try_context
|
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(
|