Spaces:

Anshu370
/

AI_Similar_Image_Visualizer

Runtime error

AI_Similar_Image_Visualizer / utility /file_handler.py

Upload 11 files

11b11b6 verified about 1 year ago

503 Bytes

from PIL import Image
from io import BytesIO
import requests
from fastapi import UploadFile
from typing import Optional

async def get_file(file: Optional[UploadFile] = None, url: Optional[str] = None) -> Image.Image:
    if file:
        image = Image.open(file.file).convert("RGB")
    elif url:
        response = requests.get(url)
        image = Image.open(BytesIO(response.content)).convert("RGB")
    else:
        raise ValueError("No image file or URL provided.")
    return image

	from PIL import Image
	from io import BytesIO
	import requests
	from fastapi import UploadFile
	from typing import Optional

	async def get_file(file: Optional[UploadFile] = None, url: Optional[str] = None) -> Image.Image:
	if file:
	image = Image.open(file.file).convert("RGB")
	elif url:
	response = requests.get(url)
	image = Image.open(BytesIO(response.content)).convert("RGB")
	else:
	raise ValueError("No image file or URL provided.")
	return image