dron3flyv3r commited on
Commit
7227cf2
·
1 Parent(s): 68e0ae6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -5,10 +5,9 @@ from PIL import Image
5
  from read import classify
6
 
7
  st.title("Pizza & Not Pizza")
8
-
9
  device = torch.device("cpu")
10
  checkpoint = torch.load(r"./best.pth.tar")
11
-
12
  model = checkpoint['model']
13
  classes = checkpoint['classes']
14
  tran = checkpoint['transform']
@@ -30,4 +29,5 @@ elif taking_picture is not None:
30
  st.write(label)
31
 
32
  else:
33
- pass
 
 
5
  from read import classify
6
 
7
  st.title("Pizza & Not Pizza")
8
+ """
9
  device = torch.device("cpu")
10
  checkpoint = torch.load(r"./best.pth.tar")
 
11
  model = checkpoint['model']
12
  classes = checkpoint['classes']
13
  tran = checkpoint['transform']
 
29
  st.write(label)
30
 
31
  else:
32
+ pass
33
+ """