Hugging Face's logo

Spaces:
Arikkod
/
FoodVisionMini
Sleeping

App Files Community
Arikkod commited on
Commit
d25de62
·
1 Parent(s): baf5230

Update model.py

Browse files
Files changed (1) hide show
  1. model.py +1 -1
model.py CHANGED
@@ -1,6 +1,6 @@
1
  import torch
2
  import torchvision
3
- from torch.nn import nn
4
 
5
  def create_effnetb2_model(num_classes:int=3, seed:int=3):
6
  weights = torchvision.models.EfficientNet_B2_Weights.DEFAULT
 
1
  import torch
2
  import torchvision
3
+ import torch.nn as nn
4
 
5
  def create_effnetb2_model(num_classes:int=3, seed:int=3):
6
  weights = torchvision.models.EfficientNet_B2_Weights.DEFAULT