Update main.py

main.py

@@ -40,7 +40,7 @@ app.mount("/static", StaticFiles(directory="static"), name="static")
 in_channels = 3
 out_channels = 7
 model = CNN(in_channels, out_channels)
-model.load_state_dict(torch.load('model/ckpt_cnn.pth', map_location=torch.device('cpu')))
+model.load_state_dict(torch.load('model/ckpt_cnn.pth', map_location=torch.device('cpu'), weights_only=True))
 model.eval()
 
 # HTML form with JavaScript for AJAX