Update app.py

app.py

@@ -22,13 +22,12 @@ def prediction(price_max,price_step,policy_net):
                 1.,     0.,   0.,   0.,   0.,   0.,   0.,   0.,   0.,   0.,   0.,   0.,   0.,   0.,   0.,   0.,   0.,   0.,   0.,   0.]
     Q_s = policy_net(to_tensor(sample_state))
     a_opt = Q_s.max(0)[1].detach()
-    st.header('Optimal price action')
-    st.write(str(price_grid[a_opt])
-    plt.figure(figsize=(16, 5))
-    plt.xlabel("Price action ($)")
-    plt.ylabel("Q ($)")
-    plt.bar(price_grid, Q_s.detach().numpy(), color='crimson',  width=6, alpha=0.8)
-    plt.show() 
+    #plt.figure(figsize=(16, 5))
+    #plt.xlabel("Price action ($)")
+    #plt.ylabel("Q ($)")
+    #plt.bar(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)