Update app.py

app.py

@@ -27,5 +27,8 @@ def prediction(price_max,price_step,policy_net):
     plt.xlabel("Price action ($)")
     plt.ylabel("Q ($)")
     plt.bar(price_grid, Q_s.detach().numpy(), color='crimson',  width=6, alpha=0.8)
-    plt.show()
-prediction(500,10,pc2)
+    plt.show() 
+st.header('Enter the Spectification')
+max_value = st.input_number('Enter the Maximun Value of Price')
+step = st.input_number('Enter the Price step')
+prediction(max_value,step,pc2)