Changed modelWeights

app.py

@@ -35,7 +35,7 @@ print(f"took {(time.time() - start_time) / 60} minutes to load face_recognition"
 #     return gr.Image(outputs[0])
 
 model_repo_id = "CSSE416-final-project/faceRecogModel"
-weight_file_id = "modelWeights100.bin"
+weight_file_id = "modelWeights101.bin"
 
 
 # 1. Load the model from Hugging Face Hub
@@ -44,7 +44,7 @@ def load_model(repo_id):
     weights_path = hf_hub_download(repo_id=model_repo_id, filename=weight_file_id)
 
     # Initialize the ResNet-18 architecture
-    model = torchvision.models.resnet18(pretrained=True) # TODO: does it matter if this is set to true or false?
+    model = torchvision.models.resnet18(pretrained=False)
     num_ftrs = model.fc.in_features
     model.fc = nn.Linear(num_ftrs, 100)  # Adjust for your task (e.g., 128 classes)
     # TODO: check if this number^^ corresponds to the number of classes