Spaces:
Sleeping
Sleeping
only for Image to Text
Browse files
app.py
CHANGED
|
@@ -18,6 +18,7 @@ from fastapi.responses import JSONResponse
|
|
| 18 |
from PIL import Image, UnidentifiedImageError
|
| 19 |
from pymongo import MongoClient
|
| 20 |
from pymongo.errors import PyMongoError, ServerSelectionTimeoutError
|
|
|
|
| 21 |
from transformers import (
|
| 22 |
AutoModelForImageTextToText,
|
| 23 |
AutoModelForSeq2SeqLM,
|
|
@@ -362,16 +363,16 @@ async def generate_caption(request: Request):
|
|
| 362 |
except Exception as exc:
|
| 363 |
raise AppError("Invalid request payload.", 422) from exc
|
| 364 |
|
| 365 |
-
uploads: list[UploadFile] = []
|
| 366 |
for key in ("files", "files[]", "file"):
|
| 367 |
for value in form.getlist(key):
|
| 368 |
-
if isinstance(value, UploadFile):
|
| 369 |
uploads.append(value)
|
| 370 |
|
| 371 |
# Fallback for clients that send non-standard multipart keys.
|
| 372 |
if not uploads:
|
| 373 |
for _, value in form.multi_items():
|
| 374 |
-
if isinstance(value, UploadFile):
|
| 375 |
uploads.append(value)
|
| 376 |
|
| 377 |
if not uploads:
|
|
|
|
| 18 |
from PIL import Image, UnidentifiedImageError
|
| 19 |
from pymongo import MongoClient
|
| 20 |
from pymongo.errors import PyMongoError, ServerSelectionTimeoutError
|
| 21 |
+
from starlette.datastructures import UploadFile as StarletteUploadFile
|
| 22 |
from transformers import (
|
| 23 |
AutoModelForImageTextToText,
|
| 24 |
AutoModelForSeq2SeqLM,
|
|
|
|
| 363 |
except Exception as exc:
|
| 364 |
raise AppError("Invalid request payload.", 422) from exc
|
| 365 |
|
| 366 |
+
uploads: list[UploadFile | StarletteUploadFile] = []
|
| 367 |
for key in ("files", "files[]", "file"):
|
| 368 |
for value in form.getlist(key):
|
| 369 |
+
if isinstance(value, (UploadFile, StarletteUploadFile)):
|
| 370 |
uploads.append(value)
|
| 371 |
|
| 372 |
# Fallback for clients that send non-standard multipart keys.
|
| 373 |
if not uploads:
|
| 374 |
for _, value in form.multi_items():
|
| 375 |
+
if isinstance(value, (UploadFile, StarletteUploadFile)):
|
| 376 |
uploads.append(value)
|
| 377 |
|
| 378 |
if not uploads:
|