Update app.py

app.py

@@ -1,5 +1,6 @@
 import streamlit as st
 import numpy as np
+import matplotlib.pyplot as plt  # Added this import statement
 import random
 
 # Streamlit app layout
@@ -43,8 +44,7 @@ if "set1" in globals():
     st.write("Scatter plot of Set 1 against Set 2")
     st.write("You can visualize the relationship between the two sets")
     st.write("Note: This plot requires 'matplotlib' which may not be supported in all Streamlit deployment environments.")
-    st.write("If the plot is not visible, please run the code locally.")
     st.pyplot(plt.scatter(set1, set2))
     
 else:
-    st.write(":red[Please specify the number of data points]")
+    st.write(":red[Please specify the number of data points]")