fix hparams

app.py

@@ -56,9 +56,24 @@ parser.add_argument('--use_crf', type=bool)
 parser.add_argument('--print_spans', action='store_true')
 args = parser.parse_args()
 
-args.num_labels = args.num_decs
-device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
+if args.task == 'seq' and args.pheno_id is not None:
+    args.num_labels = 1
+elif args.task == 'seq':
+    args.num_labels = args.num_phenos
+elif args.task == 'token':
+    if args.use_umls:
+        args.num_labels = args.num_umls_tags
+    else:
+        args.num_labels = args.num_decs
+    if args.label_encoding == 'multiclass':
+        args.num_labels = args.num_labels * 2 + 1
+    elif args.label_encoding == 'bo':
+        args.num_labels *= 2
+    elif args.label_encoding == 'boe':
+        args.num_labels *= 3
+
 
+device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 tokenizer = load_tokenizer(args.model_name)
 model = load_model(args, device)[0]
 model.eval()