mzhKU commited on
Commit
da01bf1
·
1 Parent(s): 8acae65

read token when local

Browse files
Files changed (1) hide show
  1. app.py +6 -6
app.py CHANGED
@@ -6,11 +6,13 @@ from tools.visit_webpage import VisitWebpageTool
6
  from tools.web_search import DuckDuckGoSearchTool
7
 
8
  from Gradio_UI import GradioUI
 
9
 
10
- f = open("hf_token_read.txt")
11
- d = f.read()
12
- f.close()
13
- os.environ["HF_TOKEN"] = d
 
14
 
15
  visit_webpage_tool = VisitWebpageTool()
16
  websearch_tool = DuckDuckGoSearchTool()
@@ -23,8 +25,6 @@ model = HfApiModel(
23
  custom_role_conversions=None,
24
  )
25
 
26
- # Import tool from Hub
27
-
28
  with open("prompts.yaml", 'r') as stream:
29
  prompt_templates = yaml.safe_load(stream)
30
 
 
6
  from tools.web_search import DuckDuckGoSearchTool
7
 
8
  from Gradio_UI import GradioUI
9
+ import socket
10
 
11
+ if socket.gethostname() == "LAPTOP-G1AQLNTL":
12
+ f = open("hf_token_read.txt")
13
+ d = f.read()
14
+ f.close()
15
+ os.environ["HF_TOKEN"] = d
16
 
17
  visit_webpage_tool = VisitWebpageTool()
18
  websearch_tool = DuckDuckGoSearchTool()
 
25
  custom_role_conversions=None,
26
  )
27
 
 
 
28
  with open("prompts.yaml", 'r') as stream:
29
  prompt_templates = yaml.safe_load(stream)
30