Update app.py

app.py

@@ -13,8 +13,12 @@ df = df.rename(columns={"Date": "ds", "Price": "y"})
 df.fillna(method='ffill', inplace=True)
 df.dropna(inplace=True)
 
-class CustomNeuralProphet(NeuralProphet):
-    pass
+def lr_scheduler_step(self, epoch, batch_idx, optimizer):
+    # Get the OneCycleLR scheduler
+    scheduler = self.optimizers[0].scheduler
+
+    # Call the `step` method on the scheduler
+    scheduler.step()
 
 m = CustomNeuralProphet(
     n_forecasts=30,