Update README.md

README.md

@@ -49,8 +49,8 @@ from torchvision.transforms import Compose, ToTensor, Resize, Normalize
 from utils.utils import denorm
 from model.hub import MultiInputResShiftHub
 
-model = MultiInputResShiftHub.from_pretrained("vfontech/Multiple-Input-Resshift-VFI")
-model.requires_grad_(False).cuda().eval()
+model = MultiInputResShiftHub.from_pretrained("vfontech/Multiple-Input-Resshift-VFI").cuda()
+model.eval()
 
 img0_path = r"_data\example_images\frame1.png"
 img2_path = r"_data\example_images\frame3.png"
@@ -66,7 +66,8 @@ 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
 
-img1 = model.reverse_process([img0, img2], tau)
+with torch.no_grad()
+  img1 = model.reverse_process([img0, img2], tau)
 
 plt.figure(figsize=(10, 5))
 plt.subplot(1, 3, 1)