Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -8,7 +8,7 @@ import requests
|
|
8 |
import uvicorn
|
9 |
from typing import List
|
10 |
import torch
|
11 |
-
from pdf2image import
|
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 |
|