Spaces:
Sleeping
Sleeping
File size: 978 Bytes
e9dc5bd 57b074a f56d84c e9dc5bd 02306c3 60790e5 02306c3 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 |
from fastapi import Depends, FastAPI
from .classifier.model import Model, get_model
app = FastAPI()
@app.get("/")
async def read_root(issue: str, model: Model = Depends(get_model)):
accessibility_confidence, nonaccessibility_confidence = model.predict(issue)
return {"Accessibility": str(accessibility_confidence),
"Non-accessibility": str(nonaccessibility_confidence)}
@app.post("/")
async def pred(issue: str, model: Model = Depends(get_model)):
accessibility_confidence, nonaccessibility_confidence = model.predict(issue)
return {"Accessibility": str(accessibility_confidence),
"Non-accessibility": str(nonaccessibility_confidence)}
# @app.post("/")
# async def predict(issue: str, model: Model = Depends(get_model)):
# accessibility_confidence, nonaccessibility_confidence = model.predict(issue)
# return {"Accessibility": str(accessibility_confidence),
# "Non-accessibility": str(nonaccessibility_confidence)}
|