Update newapi.py

newapi.py

@@ -8,25 +8,28 @@ from utils import BrainTumorModel, GliomaStageModel
 
 app = FastAPI()
 
-# Local model paths (files must be uploaded to the Space)
+# === Use exact filenames from the Space directory ===
 btd_model_path = "brain_tumor_model.pth"
 glioma_model_path = "glioma_stage_model.pth"
 
-# Load models
+# === Load Brain Tumor Model ===
 btd_model = BrainTumorModel()
 btd_model.load_state_dict(torch.load(btd_model_path, map_location=torch.device('cpu')))
 btd_model.eval()
 
+# === Load Glioma Stage Model ===
 glioma_model = GliomaStageModel()
 glioma_model.load_state_dict(torch.load(glioma_model_path, map_location=torch.device('cpu')))
 glioma_model.eval()
 
-# Image transformation
+# === Image Transform ===
 transform = transforms.Compose([
     transforms.Resize((224, 224)),
     transforms.ToTensor(),
 ])
 
+# === Routes ===
+
 @app.post("/predict/")
 async def predict(file: UploadFile = File(...)):
     try:
@@ -45,6 +48,7 @@ async def predict(file: UploadFile = File(...)):
     except Exception as e:
         return JSONResponse(content={"error": str(e)})
 
+
 @app.post("/glioma-stage/")
 async def glioma_stage(file: UploadFile = File(...)):
     try: