Update app.py

app.py

@@ -38,7 +38,10 @@ from urllib.request import urlretrieve
 from scipy.interpolate import LinearNDInterpolator
 from imageio import imread, imwrite
 
+device = "cuda" if torch.cuda.is_available() else "cpu"
 
+model = raft_large(weights=Raft_Large_Weights.DEFAULT, progress=False).to(device)
+model = model.eval()
 
 def write_flo(flow, filename):
     """
@@ -116,10 +119,7 @@ def infer(frameA, frameB):
     
 
 # If you can, run this example on a GPU, it will be a lot faster.
-    device = "cuda" if torch.cuda.is_available() else "cpu"
-
-    model = raft_large(weights=Raft_Large_Weights.DEFAULT, progress=False).to(device)
-    model = model.eval()
+    
 
     list_of_flows = model(img1_batch.to(device), img2_batch.to(device))
     print(f"list_of_flows type = {type(list_of_flows)}")