Spaces:
Running
on
CPU Upgrade
Running
on
CPU Upgrade
Update src/rag/pipeline.py
Browse files- src/rag/pipeline.py +1 -1
src/rag/pipeline.py
CHANGED
@@ -86,7 +86,7 @@ class RAGPipeline:
|
|
86 |
prompt_node = PromptNode(
|
87 |
model_name_or_path="gpt-4",
|
88 |
default_prompt_template=rag_prompt,
|
89 |
-
api_key=
|
90 |
max_length=3000,
|
91 |
model_kwargs={"temperature": 0.2, "max_tokens": 4096},
|
92 |
)
|
|
|
86 |
prompt_node = PromptNode(
|
87 |
model_name_or_path="gpt-4",
|
88 |
default_prompt_template=rag_prompt,
|
89 |
+
api_key=OPENAI_API_KEY,
|
90 |
max_length=3000,
|
91 |
model_kwargs={"temperature": 0.2, "max_tokens": 4096},
|
92 |
)
|