Update app.py

app.py

@@ -68,7 +68,15 @@ def solve_captcha(data_bytes: bytes = None, file_name: str = None):
         if data_bytes is None:
             print('[CAPTCHA RESOLVER NN] ПУСТОТА ВМЕСТО БАЙТОВ!')
             return None
-        img = decode_img(data_bytes)
+        if data_bytes is np.ndarray:
+            print('Img is ndarray')
+            img = decode_img_array(data_bytes)
+        else:
+            print('Img is bytes')
+            img_byte_arr = io.BytesIO()
+            inp.save(img_byte_arr, format='PNG')
+            img_byte_arr = img_byte_arr.getvalue()
+            img = decode_img(data_bytes)
         #print(img)
         pred_onx = sess.run(None, {name: img})[0]
         ans = get_result(pred_onx)
@@ -83,9 +91,7 @@ def image_classifier(inp):
     print('INPUT GOT>>>', inp)
     #gradio img inp - numpy ndarray
     
-    img_byte_arr = io.BytesIO()
-    inp.save(img_byte_arr, format='PNG')
-    img_byte_arr = img_byte_arr.getvalue()
+    
     result = solve_captcha(img_byte_arr)
     return result
 #inputs=gr.Image(type="pil")