meghanaraok commited on
Commit
23a36b4
1 Parent(s): 57f84df

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -100,7 +100,7 @@ def predict_icd(text_input, model_name, label_count):
100
  num_labels = label_dict.shape[0]
101
 
102
  if model_name != "ClinicalLongformer":
103
- config_json = "meghanaraok/"+model_name+"_"+label_count+"/config.json"
104
  parser = HfArgumentParser((ModelArguments, DataTrainingArguments, TrainingArguments))
105
  model_args, data_args, training_args = parser.parse_json_file(json_file=config_json)
106
  coding_model_config, model_class = get_coding_model_config(model_name, model_args, data_args, label_dict, label_dict.shape[0])
 
100
  num_labels = label_dict.shape[0]
101
 
102
  if model_name != "ClinicalLongformer":
103
+ config_json = "config/"+model_name+"_"+label_count+"/config.json"
104
  parser = HfArgumentParser((ModelArguments, DataTrainingArguments, TrainingArguments))
105
  model_args, data_args, training_args = parser.parse_json_file(json_file=config_json)
106
  coding_model_config, model_class = get_coding_model_config(model_name, model_args, data_args, label_dict, label_dict.shape[0])