Update app.py

app.py

@@ -134,7 +134,7 @@ class DiffusionModel(nn.Module):
         self.alphas = 1. - self.betas
         self.register_buffer('alpha_bars', torch.cumprod(self.alphas, dim=0))
 
-    @torch.no_grad()
+@torch.no_grad()
 def sample(model, num_images, timesteps, img_size, num_classes, labels, device, progress_callback=None):
     # Initialize with properly scaled noise
     x_t = torch.randn((num_images, 3, img_size, img_size), device=device) * 0.5  # Reduced initial noise scale