api.py

# =========================== # 
# Example API Usage in Python # 
# =========================== # 
 
# ======================================== # 
# Official Doc for Unity # 
# https://github.com/huggingface/unity-api # 
# ======================================== # 
 
from gradio_client import Client 
from PIL import Image 
import base64 
import io 
 
 
def main(path: str, width: int = 512, height: int = 512): 
 client = Client("Haoming02/hello-world") 
 
 img = Image.open(path) 
 
 with io.BytesIO() as buffer: 
 img.save(buffer, format="PNG") 
 base64_image = base64.b64encode(buffer.getvalue()).decode() 
 
 log, image = client.predict( 
 image=base64_image, 
 width=int(width), 
 height=int(height), 
 api_name="/resizeImage", 
 ) 
 
 print(log) 
 
 image = Image.open(io.BytesIO(base64.b64decode(image))) 
 image.save("./output.png", optimize=True) 
 
 
if __name__ == "__main__": 
 import sys 
 main(*sys.argv[1:])