Update README.md

README.md

@@ -66,8 +66,7 @@ img0 = transforms(Image.open(img0_path).convert("RGB")).unsqueeze(0).cuda()
 img2 = transforms(Image.open(img2_path).convert("RGB")).unsqueeze(0).cuda()
 tau = 0.5
 
-with torch.no_grad():
-  img1 = model.reverse_process([img0, img2], tau)
+img1 = model.reverse_process([img0, img2], tau)
 
 plt.figure(figsize=(10, 5))
 plt.subplot(1, 3, 1)