Legola commited on
Commit
e3abfd6
·
1 Parent(s): 68e3f69

Update model.py

Browse files
Files changed (1) hide show
  1. model.py +0 -5
model.py CHANGED
@@ -82,11 +82,6 @@ def image_transform(image):
82
  return image
83
 
84
 
85
- # Create EffNetB2 pretrained weights, transforms and model
86
- weights = torchvision.models.EfficientNet_B2_Weights.DEFAULT
87
- transforms = weights.transforms()
88
- model = torchvision.models.efficientnet_b2(weights=weights)
89
-
90
  #Defining a model
91
  # weights=weights=torchvision.models.VGG19_Weights.IMAGENET1K_V1
92
  # cnn = models.vgg19(weights=weights).features.eval()
 
82
  return image
83
 
84
 
 
 
 
 
 
85
  #Defining a model
86
  # weights=weights=torchvision.models.VGG19_Weights.IMAGENET1K_V1
87
  # cnn = models.vgg19(weights=weights).features.eval()