Update app.py

app.py

@@ -388,7 +388,7 @@ class MOMSystemLoop:
     def feedback(self, m_root, A_modes, Q_drive):
         decay = 0.995
         noise_level = 1e-4
-        A_modes_new = A_modes * decay + noise_level * torch.randn_like(A_modes, generator=self.gen, device=self.device)
+        A_modes_new = A_modes * decay + noise_level * torch.random_like(A_modes, generator=self.gen, device=self.device)
         A_modes_new = torch.clamp(A_modes_new, min=0.0)
         m_root_new = m_root * decay + noise_level * torch.randn_like(m_root, generator=self.gen, device=self.device)
         m_root_new = torch.clamp(m_root_new, min=0.0)