ksatzke commited on
Commit
1828ce9
·
verified ·
1 Parent(s): 9cc9a99

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +9 -2
app.py CHANGED
@@ -608,7 +608,7 @@ def compute_model_card_evaluation_results(tokenizer, model_checkpoint, raw_datas
608
  result = trainer.evaluate()
609
  return result
610
 
611
-
612
 
613
  def main():
614
  try:
@@ -622,6 +622,13 @@ def main():
622
  except Exception as e:
623
  logger.error(f"Error starting application: {e}")
624
  raise
 
 
625
 
626
  if __name__ == "__main__":
627
- main()
 
 
 
 
 
 
608
  result = trainer.evaluate()
609
  return result
610
 
611
+ """
612
 
613
  def main():
614
  try:
 
622
  except Exception as e:
623
  logger.error(f"Error starting application: {e}")
624
  raise
625
+ """
626
+
627
 
628
  if __name__ == "__main__":
629
+ model_checkpoint = model_checkpoint
630
+ raw_datasets = load_dataset(dataset_name, "mrpc")
631
+ metric = load("glue", "mrpc")
632
+ tokenizer = AutoTokenizer.from_pretrained(model_checkpoint)
633
+ output = compute_model_card_evaluation_results(tokenizer, model_checkpoint, raw_datasets, metric)
634
+ print(json.dumps(output))