Spaces:

oddadmix
/

Qari-Arabic-OCR

Running on Zero

oddadmix commited on Mar 2

Commit

13997eb

verified ·

1 Parent(s): 236a5de

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -21,6 +21,8 @@ max_tokens = 2000
 @spaces.GPU
 def perform_ocr(image):
     """Process image and extract text using OCR model"""
     image = Image.fromarray(image)
     src = str(uuid.uuid4()) + ".png"

 @spaces.GPU
 def perform_ocr(image):
+    if image == None:
+        return "Error Processing"
     """Process image and extract text using OCR model"""
     image = Image.fromarray(image)
     src = str(uuid.uuid4()) + ".png"