makanaiii commited on
Commit
4d5bb7b
·
1 Parent(s): 2910d80

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -6,8 +6,8 @@ import torchvision
6
 
7
  from torch import nn
8
  from torchvision import transforms
9
- from helper_functions import set_seeds
10
-
11
  device = "cuda" if torch.cuda.is_available() else "cpu"
12
  device
13
 
 
6
 
7
  from torch import nn
8
  from torchvision import transforms
9
+ import helper_functions
10
+ set_seeds()
11
  device = "cuda" if torch.cuda.is_available() else "cpu"
12
  device
13