Spaces:
Running
on
CPU Upgrade
Running
on
CPU Upgrade
ts startup
Browse files- app/nodes.py +1 -2
app/nodes.py
CHANGED
|
@@ -21,7 +21,6 @@ config = getconfig("params.cfg")
|
|
| 21 |
RETRIEVER = config.get("retriever", "RETRIEVER")
|
| 22 |
GENERATOR = config.get("generator", "GENERATOR")
|
| 23 |
INGESTOR = config.get("ingestor", "INGESTOR")
|
| 24 |
-
GEOJSON_INGESTOR = config.get("ingestor", "GEOJSON_INGESTOR")
|
| 25 |
MAX_CONTEXT_CHARS = int(config.get("general", "MAX_CONTEXT_CHARS"))
|
| 26 |
|
| 27 |
retriever_adapter = RetrieverAdapter("params.cfg")
|
|
@@ -65,7 +64,7 @@ def ingest_node(state: GraphState) -> GraphState:
|
|
| 65 |
|
| 66 |
try:
|
| 67 |
# Choose ingestor based on file type
|
| 68 |
-
ingestor_url =
|
| 69 |
logger.info(f"Using ingestor: {ingestor_url}")
|
| 70 |
|
| 71 |
client = Client(ingestor_url, hf_token=os.getenv("HF_TOKEN"))
|
|
|
|
| 21 |
RETRIEVER = config.get("retriever", "RETRIEVER")
|
| 22 |
GENERATOR = config.get("generator", "GENERATOR")
|
| 23 |
INGESTOR = config.get("ingestor", "INGESTOR")
|
|
|
|
| 24 |
MAX_CONTEXT_CHARS = int(config.get("general", "MAX_CONTEXT_CHARS"))
|
| 25 |
|
| 26 |
retriever_adapter = RetrieverAdapter("params.cfg")
|
|
|
|
| 64 |
|
| 65 |
try:
|
| 66 |
# Choose ingestor based on file type
|
| 67 |
+
ingestor_url = INGESTOR
|
| 68 |
logger.info(f"Using ingestor: {ingestor_url}")
|
| 69 |
|
| 70 |
client = Client(ingestor_url, hf_token=os.getenv("HF_TOKEN"))
|