Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -21,7 +21,6 @@ class Gaussian2DLayer(tf.keras.layers.Layer):
|
|
| 21 |
self.starting_position = starting_position
|
| 22 |
self.standard_deviation = standard_deviation
|
| 23 |
|
| 24 |
-
def build(self, input_shape):
|
| 25 |
# -1 --> 0.1, 0 --> 1, 1 --> 10
|
| 26 |
weights = tf.repeat([self.standard_deviation], self.num_gaussians, axis=0).numpy()
|
| 27 |
init_cov = tf.constant_initializer(weights)
|
|
|
|
| 21 |
self.starting_position = starting_position
|
| 22 |
self.standard_deviation = standard_deviation
|
| 23 |
|
|
|
|
| 24 |
# -1 --> 0.1, 0 --> 1, 1 --> 10
|
| 25 |
weights = tf.repeat([self.standard_deviation], self.num_gaussians, axis=0).numpy()
|
| 26 |
init_cov = tf.constant_initializer(weights)
|