ksatzke commited on
Commit
e6a0876
·
verified ·
1 Parent(s): bfc8e91

Update compute_model_property.py

Browse files
Files changed (1) hide show
  1. compute_model_property.py +3 -3
compute_model_property.py CHANGED
@@ -77,10 +77,10 @@ app.add_middleware(
77
  def api_home():
78
  return {'detail': 'Welcome to Bastions Model evaluation!'}
79
 
80
- @app.post("/api/evaluate/{name}", summary = "Input dataset and model identifiers", tags = ["Test API"])
81
- def return_output(name):
82
  """
83
- Some descriptive doc-string
84
  """
85
  model_checkpoint = "sgugger/glue-mrpc"
86
  dataset_name = "nyu-mll/glue"
 
77
  def api_home():
78
  return {'detail': 'Welcome to Bastions Model evaluation!'}
79
 
80
+ @app.post("/api/evaluate/{model_checkpoint, dataset_name}", summary = "Input dataset and model identifiers", tags = ["Test API"])
81
+ def return_output(model_checkpoint, dataset_name):
82
  """
83
+ Please enter model and dataset identifiers
84
  """
85
  model_checkpoint = "sgugger/glue-mrpc"
86
  dataset_name = "nyu-mll/glue"