Update app.py

app.py

@@ -46,7 +46,8 @@ def predict():
     data_df = prepare_data(date)
     if data_df.empty:
         return jsonify({'error': 'No data available for the given date'}), 400
-    top_5_stocks = predict_stock_codes(data_df)
+    y_pred = model.predict(data_df)
+    top_5_stocks = predict_stock_codes(y_pred, data_df)
     return jsonify({'top_5_stocks': top_5_stocks})
 
 if __name__ == '__main__':