Update app.py

app.py

@@ -99,10 +99,9 @@ if st.button('Analyze'):
         
         #Call the function to plot the final graph
         plot_final_graph()
-        df_test = pd.DataFrame(y_test)
-        df_predicted = pd.DataFrame(y_predicted)
+        df_test = pd.DataFrame(y_test, columns=['Original_Price'])
+        df_predicted = pd.DataFrame(y_predicted, columns=['Predicted_Price'])
         df_predictions = pd.concat([df_test, df_predicted], axis=1)
-        #df_predictions = pd.DataFrame({'Original_Price': y_test, 'Predicted_Price': y_predicted}, columns=['Original_Price', 'Predicted_Price'])
 
         st.subheader("Original Stock Pice and Predicted Stock Price:")
         st.dataframe(df_predictions,use_container_width=True)