Update newapi.py

newapi.py

@@ -5,24 +5,14 @@ from PIL import Image
 import torch
 import torchvision.transforms as transforms
 from utils import BrainTumorModel, GliomaStageModel
-from huggingface_hub import hf_hub_download
 
 app = FastAPI()
 
-# Download models from the Space's repo
-btd_model_path = hf_hub_download(
-    repo_id="Codewithsalty/brain-tumor-api",
-    filename="brain_tumor_model.pth",
-    repo_type="space"
-)
+# Local model paths (files must be uploaded to the Space)
+btd_model_path = "brain_tumor_model.pth"
+glioma_model_path = "glioma_stage_model.pth"
 
-glioma_model_path = hf_hub_download(
-    repo_id="Codewithsalty/brain-tumor-api",
-    filename="glioma_stage_model.pth",
-    repo_type="space"
-)
-
-# Load and prepare models
+# Load models
 btd_model = BrainTumorModel()
 btd_model.load_state_dict(torch.load(btd_model_path, map_location=torch.device('cpu')))
 btd_model.eval()
@@ -31,7 +21,7 @@ glioma_model = GliomaStageModel()
 glioma_model.load_state_dict(torch.load(glioma_model_path, map_location=torch.device('cpu')))
 glioma_model.eval()
 
-# Image preprocessing
+# Image transformation
 transform = transforms.Compose([
     transforms.Resize((224, 224)),
     transforms.ToTensor(),