mtyrrell commited on
Commit
f8b24dc
·
verified ·
1 Parent(s): 1b0f717

Update app/main.py

Browse files
Files changed (1) hide show
  1. 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/chatfed_retriever", hf_token=HF_TOKEN) # HF repo name
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", ""),