Update app.py

app.py

@@ -27,6 +27,7 @@ import os
 from tensorflow.keras.applications.resnet50 import ResNet50,preprocess_input, decode_predictions
 from tensorflow.keras.preprocessing import image
 from sklearn.feature_extraction.text import TfidfVectorizer
+import imageio
 
 
 
@@ -185,6 +186,7 @@ def extract_line_from_image(image, lower_line, upper_line):
     return img_copy[lower_boundary:upper_boundary, :]
 
 def extract(image):
+    image = imageio.imread(image)[:,:,:3]
     img = rgb2gray(image)
 
     #img = rgb2gray(imread("Penwritten_2048x.jpeg"))
@@ -440,6 +442,7 @@ def compute_tfidf_embeddings(documents1, documents2):
 
 
 def extract_eval(image1,image2,image3,image4):
+    print(image1)
     ideal_text=extract(image1)
     print("Extracting Ideal Text \n")
     print(ideal_text)