Update app.py

app.py

@@ -7,7 +7,7 @@ from read import classify
 st.title("Pizza & Not Pizza")
 
 device = torch.device("cpu")
-checkpoint = torch.load(r"best.pth.tar", map_location=device)
+checkpoint = torch.load(r"./best.pth.tar")
 
 model = checkpoint['model']
 classes = checkpoint['classes']
@@ -27,4 +27,7 @@ elif taking_picture is not None:
     img = Image.open(taking_picture)
     st.image(img, caption='Uploaded Image.', use_column_width=True)
     label = classify(model, img, tran, classes, device)
-    st.write(label)
+    st.write(label)
+
+else:
+    pass