Spaces:
Running
Running
meghanaraok
commited on
Commit
•
23a36b4
1
Parent(s):
57f84df
Update app.py
Browse files
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 = "
|
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])
|