fastapi

1
from enum import Enum
2

3
from fastapi import FastAPI
4

5

6
class ModelName(str, Enum):
7
    alexnet = "alexnet"
8
    resnet = "resnet"
9
    lenet = "lenet"
10

11

12
app = FastAPI()
13

14

15
@app.get("/models/{model_name}")
16
async def get_model(model_name: ModelName):
17
    if model_name is ModelName.alexnet:
18
        return {"model_name": model_name, "message": "Deep Learning FTW!"}
19

20
    if model_name.value == "lenet":
21
        return {"model_name": model_name, "message": "LeCNN all the images"}
22

23
    return {"model_name": model_name, "message": "Have some residuals"}
24