Update app.py

app.py

@@ -113,10 +113,6 @@ device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 rife_model = Model()
 rife_model.load_model("train_log", -1)
 rife_model.eval()
-rife_model.device()
-
-if torch.cuda.is_available():
-    rife_model.flownet = rife_model.flownet.half()
 
 
 @torch.no_grad()
@@ -417,6 +413,8 @@ def run_inference(
     if frame_multiplier > 1:
         start = time.time()
         print(f"Processing frames (RIFE Multiplier: {frame_multiplier}x)...")
+        rife_model.device()
+        rife_model.flownet = rife_model.flownet.half()
         final_frames = interpolate_bits(raw_frames_np, multiplier=int(frame_multiplier))
         print("Interpolation time passed:", time.time() - start)
     else: