FilesExpand file tree

 main

/

main.py

Copy path

Blame

More file actions

Blame

More file actions

Latest commit

 

History

History

History

36 lines (24 loc) · 908 Bytes

 main

/

main.py

Copy path

Top

File metadata and controls

• Code
• Blame

36 lines (24 loc) · 908 Bytes

Raw

Copy raw file

Download raw file

Open symbols panel

Edit and raw actions

1

2

3

4

5

6

7

8

9

10

11

12

13

14

15

16

17

18

19

20

21

22

23

24

25

26

27

28

29

30

31

32

33

34

35

36

from fastapi import FastAPI

from enum import Enum

class ModelName(str, Enum):

alexnet = "alexnet"

resnet = "resnet"

lenet = "lenet"

app = FastAPI()

@app.get("/search/10001")

async def default_search():

return {"default_results": "default search"}

@app.get("/search/{item_id}")

async def search(item_id: int):

item_list = [10002, 10003, 10004, 10005]

if item_list.index(item_id):

return {"result": f"Item found {item_id}"}

else:

return {"result": f"Item NOT found {item_id}"}

@app.get("/models/{model_name}")

async def get_model(model_name: ModelName):

if model_name == ModelName.alexnet:

return {"model_name": model_name, "message": "Deep Learning FTW!"}

if model_name.value == "lenet":

return {"model_name": model_name, "message": "LeCNN all the images"}

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