Update app.py

app.py

@@ -43,12 +43,12 @@ m = CustomNeuralProphet(
 # Set the custom LR scheduler
 optimizer = m.optimizer
 optimizer.lr_scheduler = {
-    "scheduler": OneCycleLR(optimizer, max_lr=0.1, steps_per_epoch=100, epochs=10),
+    "scheduler": OneCycleLR(optimizer),
     "interval": "step",
     "frequency": 1,
 }
 
-m.fit(df, freq='D', epochs=10, validate_each_epoch=True, valid_p=0.2)
+m.fit(df, freq='D')
 
 future = m.make_future_dataframe(df, periods=30, n_historic_predictions=True)
 forecast = m.predict(future)