Final

app/main.py

@@ -7,10 +7,7 @@ import torchvision.transforms.functional as tf
 from fastapi import FastAPI, UploadFile
 
 app=FastAPI(title="Alzheimer Detection API")
-@app.get("/generate")
-def display(text: str):
-    return {"yoy":text}
-"""
+
 def classify_img(model,img):
     img=tf.to_tensor(img)
     img=img.unsqueeze(0)
@@ -32,6 +29,10 @@ def get_alzheimer_model():
 
 
 
+@app.get("/")
+def display():
+    return "Welcome to Alzheimer Detection Api"
+
 @app.post("/predict")
 def predict(file: UploadFile):
     img=Image.open(file.file).convert("RGB")
@@ -39,4 +40,4 @@ def predict(file: UploadFile):
     img=np.array(img)
     model= get_alzheimer_model()
     label,probability=classify_img(model,img)
-    return {"label":label.item(),"probability":probability.item()}"""
+    return {"label":label.item(),"probability":probability.item()}