Update app.py

app.py

@@ -22,7 +22,8 @@ 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()
-    print(f'Optimal price action {price_grid[a_opt]}')
+    st.header('Optimal price action')
+    st.write(str(price_grid[a_opt])
     plt.figure(figsize=(16, 5))
     plt.xlabel("Price action ($)")
     plt.ylabel("Q ($)")