Spaces:

Hassan145
/

Vegetable_Image_Classifier

Sleeping

Hassan145 commited on May 30, 2025

Commit

5d2f608

1 Parent(s): 50098df

Your commit message

Files changed (1) hide show

model.py CHANGED Viewed

@@ -3,6 +3,7 @@
 import torch
 from torchvision import transforms, models
 import torch.nn as nn
 def create_resnet_model(num_classes: int = 4, seed: int = 42):
     transform = transforms.Compose([

 import torch
 from torchvision import transforms, models
 import torch.nn as nn
+from torchvision.models import resnet18, ResNet18_Weights
 def create_resnet_model(num_classes: int = 4, seed: int = 42):
     transform = transforms.Compose([