Update app.py

app.py

@@ -295,7 +295,7 @@ def test_model():
                 train_data = data.iloc[:split_index]
                 test_data = data.iloc[split_index:]
 
-                predictor = EnhancedStockPredictor(train_data)
+                predictor = StockPredictor(train_data)
                 predictor.preprocess_data()
                 if predictor.train_model():
                     test_pred = predictor.predict(days=len(test_data))
@@ -353,7 +353,7 @@ def predict_stock_prices():
                 st.write(data.describe())
                 st.dataframe(data.head())
 
-                predictor = EnhancedStockPredictor(data)
+                predictor = StockPredictor(data)
                 predictor.preprocess_data()
                 if predictor.train_model():
                     predictions = predictor.predict(days=days_to_predict)