Update app.py

app.py

@@ -24,7 +24,7 @@ class PortfolioEnv(Env):
 
     def step(self, action):
         action = np.clip(action, 0, 1)
-        weights = action / np.sum(action)
+        weights = action / (np.sum(action) + 1e-8)  # Prevent division by zero
 
         portfolio_return = np.dot(weights, self.returns.iloc[self.current_step].values)
         self.current_balance *= (1 + portfolio_return)