Spaces:

kby-ai
/

IDCardRecognition

Running

justin2341 commited on Jan 31

Commit

d889286

verified ·

1 Parent(s): 732ae27

Update demo.py

Files changed (1) hide show

demo.py CHANGED Viewed

@@ -5,6 +5,9 @@ from PIL import Image
 def idcard_recognition(frame1, frame2):
     url = "http://127.0.0.1:8080/idcard_recognition_multi_page"
     if frame2 is None:
         files = {'file1': open(frame1, 'rb')}

 def idcard_recognition(frame1, frame2):
+    if frame1 is None:
+        return ["Front Image Empty", None]
     url = "http://127.0.0.1:8080/idcard_recognition_multi_page"
     if frame2 is None:
         files = {'file1': open(frame1, 'rb')}