Update app.py

app.py

@@ -155,7 +155,7 @@ state = train_state.TrainState.create(
 )
 
 # Modify the train_step function
-ef train_step(state, batch, rng):
+def train_step(state, batch, rng):
     def compute_loss(params, pixel_values, rng):
         # Ensure pixel_values are float32
         pixel_values = jnp.array(pixel_values, dtype=jnp.float32)