Update compute_model_property.py

compute_model_property.py

@@ -272,9 +272,21 @@ def compute_model_card_evaluation_results(model_checkpoint, raw_datasets, metric
     result = trainer.evaluate()
     return result
 
-
-@app.get("/evaluate")
+app.add_middleware(
+    CORSMiddleware,
+    allow_origins=["*"],
+    allow_credentials=True,
+    allow_methods=["*"],
+    allow_headers=["*"],
+)
+
+@app.get("/", tags = ["Home"])
+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 = "sgugger/glue-mrpc"
     dataset_name = "nyu-mll/glue" 
     metric = ["glue", "mrpc"]