Spaces:
Build error
Build error
File size: 1,645 Bytes
e6b7912 7695918 e6b7912 7695918 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 |
import gradio as ui; import requests; from requests.exceptions import Timeout
from io import BytesIO; from PIL import Image
###################################################################################################################
# Import from modules
###################################################################################################################
from modules.service_endpoints import *
from modules.service_configs import *
###################################################################################################################
# Image upscale process
###################################################################################################################
def upscale(upscale_input, progress=ui.Progress()):
image_bytes = BytesIO()
upscale_input.save(image_bytes, format='JPEG')
print(receive())
payload = {'model_version': (None, '1')}; data = [('image',('upscale_input.jpg', image_bytes.getvalue(), 'image/jpeg'))]
try:
progress(0.99, desc="Upscaling image...")
response = requests.post(mode['upscale'], headers=head, data=payload, files=data, timeout=30)
if len(response.content) < 65 * 1024:
print(reject())
ui.Warning(message=single_error)
return None
print(done())
return Image.open(BytesIO(response.content))
except Timeout:
print(timeout())
ui.Warning(message=single_error)
return None
except Exception as e:
print(f"An error occurred: {e}")
ui.Warning(message=single_error)
return None
|