Chryslerx10 commited on
Commit
b783e79
Β·
1 Parent(s): 8e2f080

config changes

Browse files
Files changed (1) hide show
  1. app.py +5 -2
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(server_name="0.0.0.0", server_port=8585, share=True, debug=True)
 
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)