HUANG-Stephanie commited on
Commit
c5d3863
1 Parent(s): 4e3d786

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -8,7 +8,7 @@ import requests
8
  import uvicorn
9
  from typing import List
10
  import torch
11
- from pdf2image import convert_from_path
12
  from PIL import Image
13
  from torch.utils.data import DataLoader
14
  from transformers import AutoProcessor
@@ -51,7 +51,7 @@ async def index(files: List[UploadFile] = File(...)):
51
  images = []
52
  ds = []
53
  for file in files:
54
- content = file.read()
55
  pdf_image_list = convert_from_bytes(content)
56
  images.extend(pdf_image_list)
57
 
 
8
  import uvicorn
9
  from typing import List
10
  import torch
11
+ from pdf2image import convert_from_bytes
12
  from PIL import Image
13
  from torch.utils.data import DataLoader
14
  from transformers import AutoProcessor
 
51
  images = []
52
  ds = []
53
  for file in files:
54
+ content = await file.read()
55
  pdf_image_list = convert_from_bytes(content)
56
  images.extend(pdf_image_list)
57