Tschoui commited on
Commit
1578f70
·
verified ·
1 Parent(s): 7ecace2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -4
app.py CHANGED
@@ -37,16 +37,15 @@ def root():
37
  "/healthz": "GET - Health check",
38
  "/predict": "POST - Predict toxicity for SMILES",
39
  },
40
- "usage": "Send POST to /predict with {'smiles': ['your_smiles_here']} and Authorization header",
41
  }
42
 
43
 
44
  @app.get("/metadata")
45
  def metadata():
46
  return {
47
- "name": "AwesomeTox",
48
  "version": "1.0.0",
49
- "max_batch_size": 256,
50
  "tox_endpoints": [
51
  "NR-AR",
52
  "NR-AR-LBD",
@@ -74,5 +73,5 @@ def predict(request: Request):
74
  predictions = predict_func(request.smiles)
75
  return {
76
  "predictions": predictions,
77
- "model_info": {"name": "random_clf", "version": "1.0.0"},
78
  }
 
37
  "/healthz": "GET - Health check",
38
  "/predict": "POST - Predict toxicity for SMILES",
39
  },
40
+ "usage": "Send POST to /predict with {'smiles': ['your_smiles_here']}",
41
  }
42
 
43
 
44
  @app.get("/metadata")
45
  def metadata():
46
  return {
47
+ "name": "Tox21 GIN classifier",
48
  "version": "1.0.0",
 
49
  "tox_endpoints": [
50
  "NR-AR",
51
  "NR-AR-LBD",
 
73
  predictions = predict_func(request.smiles)
74
  return {
75
  "predictions": predictions,
76
+ "model_info": {"name": "Tox21 GIN classifier", "version": "1.0.0"},
77
  }