Spaces:
Running
Running
Update prompt_refiner.py
Browse files- prompt_refiner.py +1 -1
prompt_refiner.py
CHANGED
@@ -5,7 +5,7 @@ from huggingface_hub.errors import HfHubHTTPError
|
|
5 |
from variables import meta_prompts, prompt_refiner_model
|
6 |
|
7 |
class PromptRefiner:
|
8 |
-
def __init__(self, api_token: str):
|
9 |
self.client = InferenceClient(token=api_token, timeout=120)
|
10 |
self.meta_prompts = meta_prompts
|
11 |
|
|
|
5 |
from variables import meta_prompts, prompt_refiner_model
|
6 |
|
7 |
class PromptRefiner:
|
8 |
+
def __init__(self, api_token: str,meta_prompts):
|
9 |
self.client = InferenceClient(token=api_token, timeout=120)
|
10 |
self.meta_prompts = meta_prompts
|
11 |
|