ksatzke commited on
Commit
e63e630
·
verified ·
1 Parent(s): 09a0314

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -75,11 +75,13 @@ if __name__ == "__main__":
75
  #model_checkpoint = "sgugger/glue-mrpc"
76
  #dataset_name = "nyu-mll/glue"
77
 
78
- metric = load("glue", "mrpc", "sst2")
79
  tokenizer = AutoTokenizer.from_pretrained(model_checkpoint)
80
  #in_container = False
81
  model_checkpoint = model_checkpoint
 
82
  raw_datasets = load_dataset(dataset_name, "sst2")
 
83
  #metric = load("glue", "mrpc")
84
  tokenizer = AutoTokenizer.from_pretrained(model_checkpoint)
85
  output = compute_model_card_evaluation_results(tokenizer, model_checkpoint, raw_datasets, metric)
 
75
  #model_checkpoint = "sgugger/glue-mrpc"
76
  #dataset_name = "nyu-mll/glue"
77
 
78
+ metric = load("glue", "mrpc")
79
  tokenizer = AutoTokenizer.from_pretrained(model_checkpoint)
80
  #in_container = False
81
  model_checkpoint = model_checkpoint
82
+ #raw_datasets = load_dataset(dataset_name, "mrpc")
83
  raw_datasets = load_dataset(dataset_name, "sst2")
84
+
85
  #metric = load("glue", "mrpc")
86
  tokenizer = AutoTokenizer.from_pretrained(model_checkpoint)
87
  output = compute_model_card_evaluation_results(tokenizer, model_checkpoint, raw_datasets, metric)