Spaces:

sczhou
/

ProPainter

Running on A10G

sczhou commited on Jan 7

Commit

8e14395

•

1 Parent(s): c09c992

update.

Files changed (1) hide show

web-demos/hugging_face/app.py CHANGED Viewed

@@ -101,7 +101,10 @@ def get_frames_from_video(video_input, video_state):
                 else:
                     break
             t = len(frames)
-            print(f'Inp video shape: t_{t}, s_{original_h}x{original_w} to s_{target_h}x{target_w}')
     except (OSError, TypeError, ValueError, KeyError, SyntaxError) as e:
         status_ok = False
         print("read_frame_source:{} error. {}\n".format(video_path, str(e)))

                 else:
                     break
             t = len(frames)
+            if t > 0:
+                print(f'Inp video shape: t_{t}, s_{original_h}x{original_w} to s_{target_h}x{target_w}')
+            else:
+                print(f'Inp video shape: t_{t}, no input video!!!')
     except (OSError, TypeError, ValueError, KeyError, SyntaxError) as e:
         status_ok = False
         print("read_frame_source:{} error. {}\n".format(video_path, str(e)))