Update compute_model_property.py

compute_model_property.py

@@ -77,13 +77,15 @@ app.add_middleware(
 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
-    
+@app.post("/api/evaluate/{name}", summary = "Input dataset and model identifiers", tags = ["Test API"])
+def return_output(name): #model_checkpoint, dataset_name
+    """
+    Some descriptive doc-string
+    """    
     model_checkpoint = "sgugger/glue-mrpc"
     dataset_name = "nyu-mll/glue" 
 
-    print(model_checkpoint, dataset_name, metric)
+    print(model_checkpoint, dataset_name, metric, name)
     
     model_checkpoint = model_checkpoint
     raw_datasets = load_dataset(dataset_name, "mrpc")