ksatzke commited on
Commit
03e6204
·
verified ·
1 Parent(s): eabc2c4

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -8
app.py CHANGED
@@ -75,12 +75,8 @@ if __name__ == "__main__":
75
 
76
  metric = load("glue", "mrpc")
77
  tokenizer = AutoTokenizer.from_pretrained(model_checkpoint)
78
- #in_container = False
79
- model_checkpoint = model_checkpoint
80
  raw_datasets = load_dataset(dataset_name, "mrpc")
81
- #raw_datasets = load_dataset(dataset_name, "glue")
82
 
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)
86
  print(json.dumps(output))
@@ -91,7 +87,4 @@ if __name__ == "__main__":
91
 
92
  #st.text_area(label="Model Card Data:", height=500, value=json.dumps(card.data.eval_results))
93
  st.header("Model Card Evaluation Results:")
94
- st.json(card.data.eval_results, expanded=True)
95
-
96
-
97
-
 
75
 
76
  metric = load("glue", "mrpc")
77
  tokenizer = AutoTokenizer.from_pretrained(model_checkpoint)
 
 
78
  raw_datasets = load_dataset(dataset_name, "mrpc")
 
79
 
 
80
  tokenizer = AutoTokenizer.from_pretrained(model_checkpoint)
81
  output = compute_model_card_evaluation_results(tokenizer, model_checkpoint, raw_datasets, metric)
82
  print(json.dumps(output))
 
87
 
88
  #st.text_area(label="Model Card Data:", height=500, value=json.dumps(card.data.eval_results))
89
  st.header("Model Card Evaluation Results:")
90
+ st.json(card.data.eval_results, expanded=True)