baconnier commited on
Commit
c9d6ce9
·
verified ·
1 Parent(s): afbf3d7

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -25,7 +25,7 @@ class RefinementOutput(BaseModel):
25
  class PromptRefiner:
26
  def __init__(self, api_token: str,meta_prompts):
27
  self.client = InferenceClient(token=api_token, timeout=120)
28
- self.meta_prompts = {
29
  "morphosis": original_meta_prompt,
30
  "verse": new_meta_prompt,
31
  "physics": metaprompt1,
@@ -34,14 +34,14 @@ class PromptRefiner:
34
  "star": echo_prompt_refiner,
35
  "math": math_meta_prompt,
36
  "arpe": autoregressive_metaprompt
37
- }
38
 
39
- # self.meta_prompts = meta_prompts
40
 
41
  def refine_prompt(self, prompt_input: PromptInput) -> tuple:
42
  try:
43
  # Select meta prompt using dictionary instead of if-elif chain
44
- print(meta_prompts)
45
  selected_meta_prompt = self.meta_prompts.get(
46
  prompt_input.meta_prompt_choice,
47
  advanced_meta_prompt
 
25
  class PromptRefiner:
26
  def __init__(self, api_token: str,meta_prompts):
27
  self.client = InferenceClient(token=api_token, timeout=120)
28
+ '''self.meta_prompts = {
29
  "morphosis": original_meta_prompt,
30
  "verse": new_meta_prompt,
31
  "physics": metaprompt1,
 
34
  "star": echo_prompt_refiner,
35
  "math": math_meta_prompt,
36
  "arpe": autoregressive_metaprompt
37
+ }'''
38
 
39
+ self.meta_prompts = meta_prompts
40
 
41
  def refine_prompt(self, prompt_input: PromptInput) -> tuple:
42
  try:
43
  # Select meta prompt using dictionary instead of if-elif chain
44
+ # print(meta_prompts)
45
  selected_meta_prompt = self.meta_prompts.get(
46
  prompt_input.meta_prompt_choice,
47
  advanced_meta_prompt