Update app.py

app.py

@@ -116,7 +116,7 @@ def infer():
     model = raft_large(weights=Raft_Large_Weights.DEFAULT, progress=False).to(device)
     model = model.eval()
 
-    list_of_flows = model(input_frame_1.to(device), input_frame_2.to(device))
+    list_of_flows = model(img1_batch.to(device), img2_batch.to(device))
     print(f"list_of_flows type = {type(list_of_flows)}")
     print(f"list_of_flows length = {len(list_of_flows)} = number of iterations of the model")