Update app.py

app.py

@@ -119,7 +119,7 @@ class UNet(nn.Module):
         x = self.up3(x, x1, t)
         output = self.outc(x)
         return output
-device = 'cuda' if torch.cuda.is_available() else 'cpu'
+device = 'cpu'
 model = UNet(device = device).to(device)
 model.load_state_dict(torch.load('Model_Saved_States/diffusion_64.pth'))
 img_size = 64