diff --git a/bird_cnn/__pycache__/server.cpython-314.pyc b/bird_cnn/__pycache__/server.cpython-314.pyc
index 5d1ddf5..1046a08 100644
Binary files a/bird_cnn/__pycache__/server.cpython-314.pyc and b/bird_cnn/__pycache__/server.cpython-314.pyc differ
diff --git a/bird_cnn/build_models/bird_cnn b/bird_cnn/build_models/bird_cnn
new file mode 100644
index 0000000..1fc5666
Binary files /dev/null and b/bird_cnn/build_models/bird_cnn differ
diff --git a/bird_cnn/server.py b/bird_cnn/server.py
index 7a63d2a..09abdbb 100644
--- a/bird_cnn/server.py
+++ b/bird_cnn/server.py
@@ -11,7 +11,7 @@ import torch.nn.functional as F
 
 from bird_cnn import Bird_CNN
 
-SAVE_PATH = "./saved_models"
+BUILD_PATH = "./build_models"
 #IMAGE_SIZE = (1141, 850)
 IMAGE_SIZE = (300, 300)
 
@@ -36,8 +36,8 @@ transform = transforms.Compose([
 
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 
-model = Bird_CNN(c_in=3, c_hidden=15, c_out=7, kernel_size=3, img_width=IMAGE_SIZE[0], img_height=IMAGE_SIZE[1])
-full_path = os.path.join(SAVE_PATH, "bird_cnn", "bird_cnn")
+model = Bird_CNN(c_in=3, c_hidden=15, c_out=7)
+full_path = os.path.join(BUILD_PATH, "bird_cnn")
 model.load_state_dict(torch.load(full_path, weights_only=False))
 model.to(device)
 model.eval()
diff --git a/cnn_website/src/Bird_CNN.jsx b/cnn_website/src/Bird_CNN.jsx
index 35575c5..bd5a62e 100644
--- a/cnn_website/src/Bird_CNN.jsx
+++ b/cnn_website/src/Bird_CNN.jsx
@@ -2,6 +2,9 @@ import { useState } from 'react'
 import './Bird_CNN.css'
 
 function Bird_CNN() {
+    const apiUrl = process.env.NODE_ENV === "development"
+        ? "http://127.0.0.1:8000"
+        : "https://api.myapp.com";
 
     const [file, setFile] = useState(null);
     const [birdClass, setBirdClass] = useState(null);
@@ -19,7 +22,7 @@ function Bird_CNN() {
         formData.append("file", file);
 
         try {
-            const response = await fetch("http://127.0.0.1:8000/predict", {
+            const response = await fetch(`${apiUrl}/predict`, {
                 method: "POST",
                 body: formData,
             });