Update README.md
Browse files
README.md
CHANGED
|
@@ -66,7 +66,7 @@ img0 = transforms(Image.open(img0_path).convert("RGB")).unsqueeze(0).cuda()
|
|
| 66 |
img2 = transforms(Image.open(img2_path).convert("RGB")).unsqueeze(0).cuda()
|
| 67 |
tau = 0.5
|
| 68 |
|
| 69 |
-
with torch.no_grad()
|
| 70 |
img1 = model.reverse_process([img0, img2], tau)
|
| 71 |
|
| 72 |
plt.figure(figsize=(10, 5))
|
|
|
|
| 66 |
img2 = transforms(Image.open(img2_path).convert("RGB")).unsqueeze(0).cuda()
|
| 67 |
tau = 0.5
|
| 68 |
|
| 69 |
+
with torch.no_grad():
|
| 70 |
img1 = model.reverse_process([img0, img2], tau)
|
| 71 |
|
| 72 |
plt.figure(figsize=(10, 5))
|