Update app.py

app.py

@@ -293,17 +293,18 @@ def extract(image):
         #cv2.imwrite('/Users/vatsalya/Desktop/demo.jpeg',line_image)
         
 
-        im=Image.fromarray(line_image)
-        im=im.convert("L")
-        im.save("demo.jpeg")
-        print("#### Image Saved #######")
-        #('demo.png',line_image)
+        # im=Image.fromarray(line_image)
+        # im=im.convert("L")
+        # im.save("demo.jpeg")
+        # print("#### Image Saved #######")
+        #imageio.imwrite('demo.png',line_image)
 
 
 
-        image = Image.open("demo.jpeg").convert("RGB")
-        #print("Started Processing")
+        #image = Image.open("demo.jpeg").convert("RGB")
         
+        #print("Started Processing")
+        image = line_image
         pixel_values = processor(images=image, return_tensors="pt").pixel_values
 
         generated_ids = model.generate(pixel_values)