umaiku commited on
Commit
8f243dd
·
verified ·
1 Parent(s): df1b3de

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -61,6 +61,7 @@ class RetrieverTool(Tool):
61
  For more information on `huggingface_hub` Inference API support, please check the docs: https://huggingface.co/docs/huggingface_hub/v0.22.2/en/guides/inference
62
  """
63
  HF_TOKEN=os.getenv('TOKEN')
 
64
  login(HF_TOKEN)
65
 
66
  model = "meta-llama/Meta-Llama-3-8B-Instruct"
@@ -75,7 +76,7 @@ embeddings = HuggingFaceEmbeddings(model_name="sentence-transformers/paraphrase-
75
  vector_db = FAISS.load_local("faiss_index_mpnet_cos", embeddings, allow_dangerous_deserialization=True, distance_strategy=DistanceStrategy.COSINE)
76
 
77
  df = pd.read_csv("bger_cedh_db 1954-2024.csv")
78
-
79
  retriever_tool = RetrieverTool(vector_db)
80
  agent = ToolCallingAgent(tools=[retriever_tool], model=HfApiModel(model))
81
 
 
61
  For more information on `huggingface_hub` Inference API support, please check the docs: https://huggingface.co/docs/huggingface_hub/v0.22.2/en/guides/inference
62
  """
63
  HF_TOKEN=os.getenv('TOKEN')
64
+ print(HF_TOKEN)
65
  login(HF_TOKEN)
66
 
67
  model = "meta-llama/Meta-Llama-3-8B-Instruct"
 
76
  vector_db = FAISS.load_local("faiss_index_mpnet_cos", embeddings, allow_dangerous_deserialization=True, distance_strategy=DistanceStrategy.COSINE)
77
 
78
  df = pd.read_csv("bger_cedh_db 1954-2024.csv")
79
+ print(df)
80
  retriever_tool = RetrieverTool(vector_db)
81
  agent = ToolCallingAgent(tools=[retriever_tool], model=HfApiModel(model))
82