Update app.py
Browse files
    	
        app.py
    CHANGED
    
    | @@ -88,7 +88,7 @@ def read_flo_file(filename): | |
| 88 | 
             
                return data2d
         | 
| 89 |  | 
| 90 | 
             
            def warp(frame1, frame2, flo_path, blend=0.5, weights_path=None):
         | 
| 91 | 
            -
                 | 
| 92 | 
             
                flow21 = np.array(flo_path)
         | 
| 93 | 
             
                frame1pil = np.array(frame1)
         | 
| 94 | 
             
                frame1_warped21 = warp_flow(frame1pil, flow21)
         | 
|  | |
| 88 | 
             
                return data2d
         | 
| 89 |  | 
| 90 | 
             
            def warp(frame1, frame2, flo_path, blend=0.5, weights_path=None):
         | 
| 91 | 
            +
                flo_path = flo_path.cpu().data.numpy()
         | 
| 92 | 
             
                flow21 = np.array(flo_path)
         | 
| 93 | 
             
                frame1pil = np.array(frame1)
         | 
| 94 | 
             
                frame1_warped21 = warp_flow(frame1pil, flow21)
         | 
