DSXiangLi commited on
Commit
8c2caa0
1 Parent(s): b13609f
Files changed (2) hide show
  1. app.py +2 -2
  2. self/generate.py +1 -2
app.py CHANGED
@@ -4,7 +4,7 @@ import gradio as gr
4
  from ape.instance import LoadFactory
5
  from ape.prompt import MyTemplate
6
  from ape.ape import *
7
- from self.generate import init_instance
8
  from self.prompt import self_prompt
9
 
10
  with gr.Blocks(title="Automatic Prompt Engineer", theme=gr.themes.Glass()) as demo:
@@ -157,6 +157,6 @@ with gr.Blocks(title="Automatic Prompt Engineer", theme=gr.themes.Glass()) as de
157
  outputs=[self_instance])
158
 
159
  # 2. 生成
160
- generate_instruct_button.click(generate, inputs=[self_instance], outputs=[fewshot, gen_data, counter])
161
 
162
  demo.launch(show_error=True)
 
4
  from ape.instance import LoadFactory
5
  from ape.prompt import MyTemplate
6
  from ape.ape import *
7
+ from self.generate import init_instance, generate_instruction
8
  from self.prompt import self_prompt
9
 
10
  with gr.Blocks(title="Automatic Prompt Engineer", theme=gr.themes.Glass()) as demo:
 
157
  outputs=[self_instance])
158
 
159
  # 2. 生成
160
+ generate_instruct_button.click(generate_instruction, inputs=[self_instance], outputs=[fewshot, gen_data, counter])
161
 
162
  demo.launch(show_error=True)
self/generate.py CHANGED
@@ -171,12 +171,11 @@ def init_instance(seed_file, openai_key, n_human, n_machine, n_instruct, prompt)
171
  # 允许用户输入prompt修改前缀指令命令
172
  if not prompt:
173
  prompt = self_prompt
174
- print(openai_key)
175
  self_instance = SELF(seed_file.name, openai_key, n_human, n_machine, n_instruct, prompt)
176
  return self_instance
177
 
178
 
179
- def generate(self_instance):
180
  keep_instruct_data = self_instance.step()
181
 
182
  return (json.dumps(self_instance.sample_few_shot, ensure_ascii=False),
 
171
  # 允许用户输入prompt修改前缀指令命令
172
  if not prompt:
173
  prompt = self_prompt
 
174
  self_instance = SELF(seed_file.name, openai_key, n_human, n_machine, n_instruct, prompt)
175
  return self_instance
176
 
177
 
178
+ def generate_instruction(self_instance):
179
  keep_instruct_data = self_instance.step()
180
 
181
  return (json.dumps(self_instance.sample_few_shot, ensure_ascii=False),