Spaces:
Sleeping
Sleeping
Update compute_model_property.py
Browse files
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/{
|
81 |
-
def return_output(
|
82 |
"""
|
83 |
-
|
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"
|