baconnier commited on
Commit
caf3fbc
·
verified ·
1 Parent(s): 259b833

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -23,7 +23,7 @@ class RefinementOutput(BaseModel):
23
  raw_content: Optional[str] = None
24
 
25
  class PromptRefiner:
26
- def __init__(self, api_token: str):
27
  self.client = InferenceClient(token=api_token, timeout=120)
28
  '''self.meta_prompts = {
29
  "morphosis": original_meta_prompt,
@@ -341,6 +341,6 @@ autoregressive_metaprompt = os.getenv('autoregressive_metaprompt')
341
  '''
342
 
343
  if __name__ == '__main__':
344
- prompt_refiner = PromptRefiner(api_token)
345
  gradio_interface = GradioInterface(prompt_refiner,custom_css)
346
  gradio_interface.launch(share=True)
 
23
  raw_content: Optional[str] = None
24
 
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,
 
341
  '''
342
 
343
  if __name__ == '__main__':
344
+ prompt_refiner = PromptRefiner(api_token,meta_prompts)
345
  gradio_interface = GradioInterface(prompt_refiner,custom_css)
346
  gradio_interface.launch(share=True)