Update app.py

app.py

@@ -5,7 +5,7 @@ import torch.nn as neural_network_module
 class Net(neural_network_module.Module):
     def __init__(self):
         super(Net, self).__init__()
-        self.fc1 = neural_network_module.Linear(len(data[0][0]), 128)
+        self.fc1 = neural_network_module.Linear(7, 128)
         self.fc2 = neural_network_module.Linear(128, 64)
         self.fc3 = neural_network_module.Linear(64, 1)