Sol

app/main.py

@@ -7,7 +7,10 @@ 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)
@@ -27,9 +30,7 @@ def get_alzheimer_model():
     model.eval()
     return model
 
-@app.get("/generate")
-def display(text: str):
-    return {"yoy":text}
+
 
 @app.post("/predict")
 def predict(file: UploadFile):
@@ -38,4 +39,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()}"""