Update app.py

app.py

@@ -67,20 +67,20 @@ def write_flo(flow, filename):
 
     
 def infer():
-    video_url = "https://download.pytorch.org/tutorial/pexelscom_pavel_danilyuk_basketball_hd.mp4"
-    video_path = Path(tempfile.mkdtemp()) / "basketball.mp4"
-    _ = urlretrieve(video_url, video_path)
+    #video_url = "https://download.pytorch.org/tutorial/pexelscom_pavel_danilyuk_basketball_hd.mp4"
+    #video_path = Path(tempfile.mkdtemp()) / "basketball.mp4"
+    #_ = urlretrieve(video_url, video_path)
    
-    frames, _, _ = read_video(str(video_path), output_format="TCHW")
+    frames, _, _ = read_video(str("./spacex.mp4"), output_format="TCHW")
     print(f"FRAME BEFORE stack: {frames[100]}")
     
-    input_frame_1 = read_image(str("./frame1.jpg"), ImageReadMode.UNCHANGED)
-    print(f"FRAME 1: {input_frame_1}")
-    input_frame_2 = read_image(str("./frame2.jpg"), ImageReadMode.UNCHANGED)
-    print(f"FRAME 1: {input_frame_2}")
+    #input_frame_1 = read_image(str("./frame1.jpg"), ImageReadMode.UNCHANGED)
+    #print(f"FRAME 1: {input_frame_1}")
+    #input_frame_2 = read_image(str("./frame2.jpg"), ImageReadMode.UNCHANGED)
+    #print(f"FRAME 1: {input_frame_2}")
     
-    img1_batch = torch.stack([input_frame_1])
-    img2_batch = torch.stack([input_frame_2])
+    img1_batch = torch.stack([frames[0]])
+    img2_batch = torch.stack([frames[1]])
     
     print(f"FRAME AFTER stack: {img1_batch}")