Update compute_model_property.py

compute_model_property.py

@@ -78,7 +78,7 @@ def api_home():
     return {'detail': 'Welcome to Bastions Model evaluation!'}
 
 @app.post("/api/evaluate", summary = "Input dataset and model identifiers", tags = ["Test API"])
-def return_output(model_checkpoint, dataset_name):
+def return_output(): #model_checkpoint, dataset_name
     
     model_checkpoint = "sgugger/glue-mrpc"
     dataset_name = "nyu-mll/glue" 
@@ -87,7 +87,6 @@ def return_output(model_checkpoint, dataset_name):
     
     model_checkpoint = model_checkpoint
     raw_datasets = load_dataset(dataset_name, "mrpc")
-    #metric = load("glue", "mrpc")
     #tokenizer = AutoTokenizer.from_pretrained(model_checkpoint)
     output = compute_model_card_evaluation_results(model_checkpoint, raw_datasets, metric)