Update app.py

app.py

@@ -19,7 +19,7 @@ class Net(neural_network_module.Module):
         
 # Load the model
 model = Net()
-model.save_state_dict(torch.load('model.pth'))
+model.load_state_dict(torch.load('model.pth'))
 
 agents = [
     'Brimstone',