Update app.py

app.py

@@ -10,7 +10,8 @@ os.environ['USE_TORCH'] = '1'
 import matplotlib.pyplot as plt
 from doctr.io import DocumentFile
 from doctr.models import ocr_predictor
-
+from PIL import Image
+import numpy as np
 # util functions
 
 def prepreocess(text):
@@ -64,8 +65,9 @@ predictor = ocr_predictor(pretrained=True)
 
 def proceed(image):
     print("image: ", image)
-
-    doc = DocumentFile.from_images(image) #reading input image
+    im = Image.fromarray(np.array(image))
+    im.save("temp.jpg")
+    doc = DocumentFile.from_images("temp.jpg") #reading input image
     result = predictor(doc) # extract words
     json_export = result.export() # convert to json
     out_words = [[prepreocess(w['value']) for w in line['words'] ] for line in json_export['pages'][0]['blocks'][0]['lines']] # extracting words