mzhKU commited on
Commit
692c3da
·
1 Parent(s): da01bf1

Read space from env

Browse files
Files changed (1) hide show
  1. app.py +4 -1
app.py CHANGED
@@ -8,12 +8,15 @@ from tools.web_search import DuckDuckGoSearchTool
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()
19
  final_answer = FinalAnswerTool()
 
8
  from Gradio_UI import GradioUI
9
  import socket
10
 
11
+ if os.getenv('HF_SPACE') == "HF_SPACE":
12
+ token = os.getenv('HF_READ_First_Agent')
13
+ else:
14
  f = open("hf_token_read.txt")
15
  d = f.read()
16
  f.close()
17
  os.environ["HF_TOKEN"] = d
18
 
19
+
20
  visit_webpage_tool = VisitWebpageTool()
21
  websearch_tool = DuckDuckGoSearchTool()
22
  final_answer = FinalAnswerTool()