Update app.py

app.py

@@ -375,7 +375,7 @@ def test_model():
                 train_data = data.iloc[:split_index]
                 test_data = data.iloc[split_index:]
 
-                predictor = StockPredictor(train_data)
+                predictor = StockPredictor(train_data)  # Updated: removed model_type argument
                 predictor.preprocess_data()
                 if predictor.train_model():
                     test_pred = predictor.predict(days=len(test_data))
@@ -417,7 +417,7 @@ def predict_stock_prices():
 
                 st.markdown(get_table_download_link(data), unsafe_allow_html=True)
 
-                predictor = StockPredictor(data)
+                predictor = StockPredictor(data)  # Updated: removed model_type argument
                 predictor.preprocess_data()
                 if predictor.train_model():
                     predictions = predictor.predict(days=days_to_predict)