jbbove commited on
Commit
8196a62
·
verified ·
1 Parent(s): a17cecc

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -1
app.py CHANGED
@@ -3,12 +3,15 @@ import datetime
3
  import requests
4
  import pytz
5
  import yaml
 
6
  from tools.final_answer import FinalAnswerTool
7
 
8
  from Gradio_UI import GradioUI
9
 
10
  API_URL = "https://api-inference.huggingface.co/models/mistralai/Mistral-7B-Instruct-v0.2"
11
- headers = {"Authorization": "Bearer HF_TOKEN"}
 
 
12
 
13
  @tool
14
  def extract_knowledge_graph(text: str) -> str:
 
3
  import requests
4
  import pytz
5
  import yaml
6
+ import os
7
  from tools.final_answer import FinalAnswerTool
8
 
9
  from Gradio_UI import GradioUI
10
 
11
  API_URL = "https://api-inference.huggingface.co/models/mistralai/Mistral-7B-Instruct-v0.2"
12
+ headers = {
13
+ "Authorization": f"Bearer {os.environ['HF_TOKEN']}",
14
+ }
15
 
16
  @tool
17
  def extract_knowledge_graph(text: str) -> str: