Update app.py

app.py

@@ -199,19 +199,20 @@ def predict_and_plot(timeframe, limit, epsilon, n_steps, ma):
 
 def predict_both_plots(limit, epsilon, n_steps, ma):
     period = f'{limit}d'
-    btc_data_1d = fetch_yfinance_data('BTC/USDT', period, timeframe='1d')
-    btc_data_4h = fetch_yfinance_data('BTC/USDT', period, timeframe='4h')
-    bch_data_1d = fetch_yfinance_data('BCH/USDT', period, timeframe='1d')
-    bch_data_4h = fetch_yfinance_data('BCH/USDT', period, timeframe='4h')
+    btc_data_1d = fetch_yfinance_data('BTC/USDT', period, interval='1d')
+    btc_data_4h = fetch_yfinance_data('BTC/USDT', period, interval='4h')
+    bch_data_1d = fetch_yfinance_data('BCH/USDT', period, interval='1d')
+    bch_data_4h = fetch_yfinance_data('BCH/USDT', period, interval='4h')
 
     btc_data_1d, _ = normalize(btc_data_1d)
     bch_data_1d, _ = normalize(bch_data_1d)
     btc_data_4h, _ = normalize(btc_data_4h)
     bch_data_4h, _ = normalize(bch_data_4h)
-    preds_1d = predictions(model_n1d_cat, btc_data_1d, bch_data_1d, name=timeframe, n_steps=n_steps)
-    preds_4h = predictions(model_n4h_cat, btc_data_4h, bch_data_4h, name=timeframe, n_steps=n_steps)
-    fig_1d = plot(preds_1d, label=label, timeframe=timeframe, ma=ma, n_steps=n_steps)
-    fig_4h = plot(preds_4h, label=label, timeframe=timeframe, ma=ma, n_steps=n_steps)
+
+    preds_1d, label_1d = predictions(model_n1d_cat, btc_data_1d, bch_data_1d, name='1 day', n_steps=n_steps)
+    preds_4h, label_4h = predictions(model_n4h_cat, btc_data_4h, bch_data_4h, name='4 hours', n_steps=n_steps)
+    fig_1d = plot(preds_1d, label=label_1d, timeframe=timeframe, ma=ma, n_steps=n_steps)
+    fig_4h = plot(preds_4h, label=label_4h, timeframe=timeframe, ma=ma, n_steps=n_steps)
     return fig_1d, fig_2g
 
 with gr.Blocks() as demo:
@@ -235,4 +236,4 @@ with gr.Blocks() as demo:
         outputs=[plot1, plot2]
     )
 
-demo.launch()
+demo.launch()