Spaces:
Sleeping
Sleeping
File size: 799 Bytes
43c5517 |
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 |
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 |