Update app.py
Browse files
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,
|