Update app.py

app.py

@@ -75,12 +75,12 @@ if __name__ == "__main__":
             #model_checkpoint = "sgugger/glue-mrpc"
             #dataset_name = "nyu-mll/glue"
     
-            metric = load("glue", "mrpc")
+            metric = load("glue", "mrpc", "sst2")
             tokenizer = AutoTokenizer.from_pretrained(model_checkpoint)
             #in_container = False
             model_checkpoint = model_checkpoint
-            raw_datasets = load_dataset(dataset_name, "mrpc")
-            metric = load("glue", "mrpc")
+            raw_datasets = load_dataset(dataset_name, "sst2")
+            #metric = load("glue", "mrpc")
             tokenizer = AutoTokenizer.from_pretrained(model_checkpoint)
             output = compute_model_card_evaluation_results(tokenizer, model_checkpoint, raw_datasets, metric)
             print(json.dumps(output))