Fix jax.numpy typos errors

app.py

@@ -21,9 +21,8 @@ noise_standard_deviation = st.sidebar.slider('Standard deviation of the noise',
 cost_function = st.sidebar.radio('What cost function you want to use for the fitting?', options=('RMSE-Loss', 'Huber-Loss'))
 
 # Generate random data
-X = np.column_stack((np.ones(number_of_observations), 
-                     np.random.random(number_of_observations)))      
-
+X = np.column_stack((jnp.ones(number_of_observations),
+                     jax.random.uniform(key, shape=(number_of_observations,), minval=0., maxval=1.))
 w = np.array([3.0, -20.0, 32.0])  # coefficients                                    
 X = np.column_stack((X, X[:,1] ** 2))   # add x**2 column
 additional_noise = 8 * jax.random.bernoulli(key, p=0.08, shape=[number_of_observations,])