Update app.py

app.py

@@ -23,8 +23,10 @@ class Net(keras.Model):
 
     def build(self, input_shape):
         self.fc1.build(input_shape)
-        self.fc2.build(self.fc1.output_shape)
-        self.fc3.build(self.fc2.output_shape)
+        output_shape = self.fc1.compute_output_shape(input_shape)
+        self.fc2.build(output_shape)
+        output_shape = self.fc2.compute_output_shape(output_shape)
+        self.fc3.build(output_shape)
         self.built = True
 
     def call(self, x):