Spaces:
Running
Running
Update app.py
Browse files
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']}
|
| 41 |
}
|
| 42 |
|
| 43 |
|
| 44 |
@app.get("/metadata")
|
| 45 |
def metadata():
|
| 46 |
return {
|
| 47 |
-
"name": "
|
| 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": "
|
| 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 |
}
|