modified code

src/resnet_model.py

@@ -7,8 +7,7 @@ class MonkeyResNet(nn.Module):
         super(MonkeyResNet, self).__init__()
         
         # Load pretrained ResNet18 model from torchvision
-        self.model = models.resnet18(pretrained=True)  # pretrained weights from ImageNet
-
+        self.model = models.resnet18(pretrained=False)
         # Replace the final fully connected layer to match number of classes
         in_features = self.model.fc.in_features
         self.model.fc = nn.Linear(in_features, num_classes)

src/streamlit_app.py

@@ -18,16 +18,16 @@ st.set_page_config(page_title="ApexID: Monkey Species Classifier", layout="wide"
 MODEL_PATH = "monkey_resnet.pth"
 CLASS_NAMES = ['n0', 'n1', 'n2', 'n3', 'n4', 'n5', 'n6', 'n7', 'n8', 'n9']
 LABEL_MAP = {
-    'n0': 'alouatta_palliata',
-    'n1': 'erythrocebus_patas',
-    'n2': 'cacajao_calvus',
-    'n3': 'macaca_fuscata',
-    'n4': 'cebuella_pygmea',
-    'n5': 'cebus_capucinus',
-    'n6': 'mico_argentatus',
-    'n7': 'saimiri_sciureus',
-    'n8': 'aotus_nigriceps',
-    'n9': 'trachypithecus_johnii'
+    'n0': 'Alouatta Palliata',
+    'n1': 'Erythrocebus Patas',
+    'n2': 'Cacajao Calvus',
+    'n3': 'Macaca Fuscata',
+    'n4': 'Cebuella Pygmea',
+    'n5': 'Cebus Capucinus',
+    'n6': 'Mico Argentatus',
+    'n7': 'Saimiri Sciureus',
+    'n8': 'Aotus Nigriceps',
+    'n9': 'Trachypithecus Johnii'
 }
 
 # Load model with caching to avoid reloading every time