Update model.py

model.py

@@ -8,7 +8,7 @@ def create_model(num_classes=7):
     model = torchvision.models.efficientnet_b2(weights=weights)
 
     for param in model.parameters():
-        param.requires_grad = False  # Freeze for inference
+        param.requires_grad = False  
 
     model.classifier = nn.Sequential(
         nn.Dropout(p=0.3),