thomasinovic commited on
Commit
f78298e
·
1 Parent(s): 48567c5

fix imports

Browse files
Files changed (2) hide show
  1. app.py +1 -2
  2. requirements.txt +2 -1
app.py CHANGED
@@ -3,8 +3,7 @@ import torch
3
  import torch.nn.functional as F
4
  import torchvision.transforms as T
5
  import json
6
- import os
7
- from pathlib import Path
8
  from CNN import CNN
9
 
10
  # def greet(name):
 
3
  import torch.nn.functional as F
4
  import torchvision.transforms as T
5
  import json
6
+
 
7
  from CNN import CNN
8
 
9
  # def greet(name):
requirements.txt CHANGED
@@ -1,2 +1,3 @@
1
  torch
2
- json
 
 
1
  torch
2
+ json
3
+ torchvision