Spaces:
Running
Running
Update app.py
Browse files
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 |
-
|
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
|