Spaces:
Runtime error
Runtime error
Commit
Β·
b783e79
1
Parent(s):
8e2f080
config changes
Browse files
app.py
CHANGED
|
@@ -11,6 +11,9 @@ from interface import create_demo
|
|
| 11 |
from medrax.agent import *
|
| 12 |
from medrax.tools import *
|
| 13 |
from medrax.utils import *
|
|
|
|
|
|
|
|
|
|
| 14 |
|
| 15 |
warnings.filterwarnings("ignore")
|
| 16 |
logging.set_verbosity_error()
|
|
@@ -121,11 +124,11 @@ if __name__ == "__main__":
|
|
| 121 |
model_dir=os.getenv("MODEL_DIR"), # Change this to the path of the model weights
|
| 122 |
temp_dir="temp", # Change this to the path of the temporary directory
|
| 123 |
device="cuda", # Change this to the device you want to use
|
| 124 |
-
model="openai/gpt-4o", # Change this to the model you want to use, e.g. gpt-4o-mini
|
| 125 |
temperature=0.7,
|
| 126 |
top_p=0.95,
|
| 127 |
openai_kwargs=openai_kwargs
|
| 128 |
)
|
| 129 |
demo = create_demo(agent, tools_dict)
|
| 130 |
|
| 131 |
-
demo.launch(
|
|
|
|
| 11 |
from medrax.agent import *
|
| 12 |
from medrax.tools import *
|
| 13 |
from medrax.utils import *
|
| 14 |
+
from huggingface_hub import login
|
| 15 |
+
|
| 16 |
+
login(token=os.getenv("HF_TOKEN"))
|
| 17 |
|
| 18 |
warnings.filterwarnings("ignore")
|
| 19 |
logging.set_verbosity_error()
|
|
|
|
| 124 |
model_dir=os.getenv("MODEL_DIR"), # Change this to the path of the model weights
|
| 125 |
temp_dir="temp", # Change this to the path of the temporary directory
|
| 126 |
device="cuda", # Change this to the device you want to use
|
| 127 |
+
model="openai/gpt-4o-mini", # Change this to the model you want to use, e.g. gpt-4o-mini
|
| 128 |
temperature=0.7,
|
| 129 |
top_p=0.95,
|
| 130 |
openai_kwargs=openai_kwargs
|
| 131 |
)
|
| 132 |
demo = create_demo(agent, tools_dict)
|
| 133 |
|
| 134 |
+
demo.launch(server_port=8585, share=True, debug=True, ssr_mode=False)
|