ahmedzein commited on
Commit
a3691b9
·
verified ·
1 Parent(s): 2663b22

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +3 -3
main.py CHANGED
@@ -34,9 +34,9 @@ async def extract_table_data(image: UploadFile = File(...)):
34
 
35
  # Open image in memory
36
  image = Image.open(io.BytesIO(image_data))
37
- rgb_img = image.convert("RGB")
38
- rgb_img.save('output.jpg')
39
- image = Image.open('output.jpg')
40
 
41
  table_fram= inference(image)
42
  if table_fram.empty:
 
34
 
35
  # Open image in memory
36
  image = Image.open(io.BytesIO(image_data))
37
+ image = image.convert("RGB")
38
+ #rgb_img.save('output.jpg')
39
+ #image = Image.open('output.jpg')
40
 
41
  table_fram= inference(image)
42
  if table_fram.empty: