Spaces:
Sleeping
Sleeping
Update initialize.py
Browse files- initialize.py +1 -1
initialize.py
CHANGED
@@ -38,7 +38,7 @@ def define_components(document_store, api_key):
|
|
38 |
token=Secret.from_token(api_key)
|
39 |
)
|
40 |
generator.warm_up()
|
41 |
-
return
|
42 |
|
43 |
def define_pipeline(retreiver, prompt_builder, generator):
|
44 |
basic_rag_pipeline = Pipeline()
|
|
|
38 |
token=Secret.from_token(api_key)
|
39 |
)
|
40 |
generator.warm_up()
|
41 |
+
return retriever, prompt_builder, generator
|
42 |
|
43 |
def define_pipeline(retreiver, prompt_builder, generator):
|
44 |
basic_rag_pipeline = Pipeline()
|