Upload model.py with huggingface_hub

model.py

@@ -13,7 +13,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)
+    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)