Spaces:
Running
Running
Update app/main.py
Browse files- app/main.py +1 -1
app/main.py
CHANGED
@@ -21,7 +21,7 @@ class GraphState(TypedDict):
|
|
21 |
|
22 |
# node 2: retriever
|
23 |
def retrieve_node(state: GraphState) -> GraphState:
|
24 |
-
client = Client("giz/
|
25 |
context = client.predict(
|
26 |
query=state["query"],
|
27 |
reports_filter=state.get("reports_filter", ""),
|
|
|
21 |
|
22 |
# node 2: retriever
|
23 |
def retrieve_node(state: GraphState) -> GraphState:
|
24 |
+
client = Client("giz/chatfed_retriever_old", hf_token=HF_TOKEN) # HF repo name
|
25 |
context = client.predict(
|
26 |
query=state["query"],
|
27 |
reports_filter=state.get("reports_filter", ""),
|