Spaces:
Running
Running
File size: 1,285 Bytes
eadd7ea 25dc585 5599066 68c2b09 152b5cd eadd7ea 5079107 eadd7ea 5079107 ffd3672 eadd7ea 25dc585 152b5cd 68c2b09 152b5cd 68c2b09 152b5cd 0a68816 152b5cd 68c2b09 152b5cd 68c2b09 152b5cd eadd7ea 5079107 87d0fd1 eadd7ea |
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 |
import gradio as gr
from gradio_client import Client, handle_file
from PIL import Image
from io import BytesIO
import os
import tempfile
def upscale_image(url):
client = Client("doevent/Face-Real-ESRGAN")
result = client.predict(
image=handle_file(url), # Directly pass the image
size="4x",
api_name="/predict"
)
# print("\nTask Completed!")
# return result
# Read the image from the file path returned by the model
if os.path.exists(result):
with open(result, 'rb') as img_file:
img_data = img_file.read()
# Convert result to PNG
img = Image.open(BytesIO(img_data))
# Save the converted image to a temporary file
with tempfile.NamedTemporaryFile(delete=False, suffix='.jpg') as temp_file:
img.save(temp_file, format="JPEG", quality=95)
temp_file_path = temp_file.name
# Optionally, delete the temp file after processing (you can remove this line if not needed)
os.remove(result)
print("\nTask Completed!")
return temp_file_path
app = gr.Interface(upscale_image,
inputs = [gr.Textbox(label="Url")],
outputs = [gr.Image(label="Upscaled Image", format='png')])
app.launch(debug=True)
|