dron3flyv3r commited on
Commit
8c92bed
·
1 Parent(s): 7227cf2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -5,7 +5,7 @@ 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
  model = checkpoint['model']
@@ -30,4 +30,3 @@ elif taking_picture is not None:
30
 
31
  else:
32
  pass
33
- """
 
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']
 
30
 
31
  else:
32
  pass