Dhruv-Ty commited on
Commit
8c13722
·
1 Parent(s): eb57a64

model-weights path fix

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -127,7 +127,7 @@ if __name__ == "__main__":
127
  agent, tools_dict = initialize_agent(
128
  "medrax/docs/system_prompts.txt",
129
  tools_to_use=selected_tools,
130
- model_dir="/model-weights", # Change this to the path of the model weights
131
  temp_dir="temp", # Change this to the path of the temporary directory
132
  device="cuda", # Change this to the device you want to use
133
  model="gpt-4o", # Change this to the model you want to use, e.g. gpt-4o-mini
 
127
  agent, tools_dict = initialize_agent(
128
  "medrax/docs/system_prompts.txt",
129
  tools_to_use=selected_tools,
130
+ model_dir="./model-weights", # Change this to the path of the model weights
131
  temp_dir="temp", # Change this to the path of the temporary directory
132
  device="cuda", # Change this to the device you want to use
133
  model="gpt-4o", # Change this to the model you want to use, e.g. gpt-4o-mini