File size: 549 Bytes
bfbce39
 
9f7b5d5
bfbce39
 
 
 
4b218fe
 
d805bce
9f7b5d5
15ec542
bfbce39
 
 
 
 
 
15ec542
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
import numpy as np
from PIL import Image
from fastapi import FastAPI, UploadFile
from utils import classify_img,get_alzheimer_model

app=FastAPI(title="Alzheimer Detection API")

@app.get("/generate")
def display(text: str):
    return {"yoy":text}

"""@app.post("/predict")
def predict(file: UploadFile):
    img=Image.open(file.file).convert("RGB")
    img=img.resize(480,480)
    img=np.array(img)
    model= get_alzheimer_model()
    label,probability=classify_img(model,img)
    return {"label":label.item(),"probability":probability.item()}"""