Views
No views yet
1def predict(path_to_image, prompt):
2 with open(path_to_image, "rb") as i:
3 b64 = base64.b64encode(i.read()).decode()
4 payload = {
5 "inputs": b64,
6 "parameters": {
7 "prompt": prompt
8 }
9 }
10 response = r.post(
11 ENDPOINT_URL, json=payload, headers={"Content-Type": "application/json"}
12 )
13 return response.json()1resp = predict(
2 path_to_image="car.png",
3 prompt="make the car green"
4)
5img = Image.open(BytesIO(base64.b64decode(resp)))