Update app.py

app.py

@@ -2,8 +2,9 @@ from flask import Flask, request, jsonify
 from sklearn.preprocessing import MinMaxScaler
 import pandas as pd
 import os
+from app2 import predict_stock_codes
 
-app = app2.py
+app = Flask(__name__)
 
 # Load the prediction model
 model = CustomModel()
@@ -45,9 +46,8 @@ def predict():
     data_df = prepare_data(date)
     if data_df.empty:
         return jsonify({'error': 'No data available for the given date'}), 400
-    y_pred = model.predict(data_df)
-    top_5_stocks = data_df.iloc[y_pred.argsort()[-5:]]
-    return jsonify({'top_5_stocks': top_5_stocks['code'].tolist()})
+    top_5_stocks = predict_stock_codes(data_df)
+    return jsonify({'top_5_stocks': top_5_stocks})
 
 if __name__ == '__main__':
     app.run(debug=True)