vidhi0405 commited on
Commit
e987372
·
1 Parent(s): 57779c1

only for Image to Text

Browse files
Files changed (1) hide show
  1. app.py +4 -3
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: