Spaces:
Sleeping
Sleeping
Update model.py
Browse filesremoved faulty import
model.py
CHANGED
|
@@ -2,7 +2,6 @@ import torch
|
|
| 2 |
import torchvision
|
| 3 |
|
| 4 |
from torch import nn
|
| 5 |
-
from helper_functions import set_seeds
|
| 6 |
|
| 7 |
def create_effnetb2_model(output_classes:int=3,
|
| 8 |
seed=42):
|
|
|
|
| 2 |
import torchvision
|
| 3 |
|
| 4 |
from torch import nn
|
|
|
|
| 5 |
|
| 6 |
def create_effnetb2_model(output_classes:int=3,
|
| 7 |
seed=42):
|