modeltest / example_client.py
yaseengoldfinchpc's picture
Git Push
43c5517
raw
history blame
799 Bytes
import requests
import base64
from PIL import Image
import io
def call_inpaint_api(image_path, mask_path, prompt):
# Update this with your actual space URL after deployment
url = "https://your-username-your-space-name.hf.space/inpaint"
files = {
'image': open(image_path, 'rb'),
'mask': open(mask_path, 'rb')
}
data = {
'prompt': prompt
}
response = requests.post(url, files=files, data=data)
if response.status_code == 200:
# Decode base64 image
img_data = base64.b64decode(response.json()['image'])
img = Image.open(io.BytesIO(img_data))
img.save('result.png')
return 'result.png'
else:
print(f"Error: {response.text}")
return None