Update app.py

app.py

@@ -28,11 +28,7 @@ def prediction(price_max,price_step,policy_net):
 def fun():
     st.header("Optimal Price Action")
     st.subheader(str(a))
-    plt.figure(figsize=(16, 5))
-    plt.xlabel("Price action ($)")
-    plt.ylabel("Q ($)")
-    plt.bar(a, b, color='crimson',  width=6, alpha=0.8)
-    plt.show() 
+    
     return 
 st.header("Enter the Specification")
 max_value = st.number_input('Enter the Maximum Value of Price',min_value=50,value = 500,step=1)
@@ -40,6 +36,11 @@ step = st.number_input('Enter the Price step',min_value = 10,value = 10,step=1)
 a,b = prediction(max_value,step,pc2)
 if st.button('Predict'):
     fun()
+    plt.figure(figsize=(16, 5))
+    plt.xlabel("Price action ($)")
+    plt.ylabel("Q ($)")
+    plt.bar(a, b, color='crimson',  width=6, alpha=0.8)
+    plt.show()