Daease commited on
Commit
c740290
·
verified ·
1 Parent(s): 8e9f79e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +8 -0
app.py CHANGED
@@ -14,10 +14,18 @@ from medrax.agent import *
14
  from medrax.tools import *
15
  from medrax.utils import *
16
 
 
 
 
17
  warnings.filterwarnings("ignore")
18
  logging.set_verbosity_error()
19
  _ = load_dotenv()
20
 
 
 
 
 
 
21
  def initialize_agent(
22
  prompt_file,
23
  tools_to_use=None,
 
14
  from medrax.tools import *
15
  from medrax.utils import *
16
 
17
+ from huggingface_hub import login
18
+ import os
19
+
20
  warnings.filterwarnings("ignore")
21
  logging.set_verbosity_error()
22
  _ = load_dotenv()
23
 
24
+
25
+ hf_token = os.getenv("HUGGINGFACE_TOKEN")
26
+ if hf_token:
27
+ login(token=hf_token)
28
+
29
  def initialize_agent(
30
  prompt_file,
31
  tools_to_use=None,