Update app.py

app.py

@@ -25,12 +25,13 @@ def prediction(price_max,price_step,policy_net):
     #plt.figure(figsize=(16, 5))
     #plt.xlabel("Price action ($)")
     #plt.ylabel("Q ($)")
-    s#t.bar_chart(price_grid, Q_s.detach().numpy(), color='crimson',  width=6, alpha=0.8)
+    #t.bar_chart(price_grid, Q_s.detach().numpy(), color='crimson',  width=6, alpha=0.8)
     #plt.show() 
     return price_grid[a_opt]
 st.header('Enter the Spectification')
 max_value = st.number_input('Enter the Maximun Value of Price',min_value=50,value = 500,step=1)
 step = st.number_input('Enter the Price step',min_value = 10,value = 10,step=1)
-a = prediction(max_value,step,pc2)
-st.header('Optimal Price Action')
-st.subheader(str(a))
+a = prediction(max_value,step,pc2
+if st.button('Predict'):
+    st.header('Optimal Price Action')
+    st.subheader(str(a))