Add test script

testing/README.md

@@ -0,0 +1,10 @@
+# Testing
+This test is to run a custom model on Hugging Face using a python script to produce an 'image.png' file. The script requires there to be two files in the same local directory: `New hairstyle applied.png` and `mask-inverted.png`. The first file is a selfie with a roughly overlaid hairstyle. The second is a mask that specifies the strength of the inpainting in particular areas in the image. I included some sample files, but you can try different ones as well.
+
+- Via a Terminal, navigate into the directory with `test-hf.py`, then run these commands:
+- `python3 -m venv venv` # This creates a "venv" - a special area to install dependencies
+- `. venv/bin/activate` # This activates the "venv" so when you install dependencies they go into the venv, instead of installing system-wide.
+- `pip install requests` # This installs the "requests" library, that helps with calling remote web services
+- `python3 test-hf.py` # This calls the endpoint on huggingface
+- You should now have an `image.png` that shows the inpainted hairstyle.
+

testing/test-hf.py

@@ -0,0 +1,31 @@
+import base64
+import requests
+import json
+
+with open("New hairstyle applied.png", "rb") as image_file:
+    image_base64 = base64.b64encode(image_file.read())
+with open("mask-inverted.png", "rb") as mask_file:
+    mask_base64 = base64.b64encode(mask_file.read())
+
+params = {"inputs": {"image": str(image_base64, encoding="utf-8"), "mask": str(mask_base64, encoding="utf-8")}}
+#print(f"input params: {params}")
+
+#url = 'https://vibg4peqt9go3gqx.us-east-1.aws.endpoints.huggingface.cloud'
+#url = 'http://localhost:5001'
+url = 'https://gcjls79f021eqt9p.us-east-1.aws.endpoints.huggingface.cloud'
+token = ''
+
+r = requests.post(url, data=json.dumps(params), headers={"Authorization": f"Bearer {token}", "Content-type": "application/json"})
+
+print(f"Status: {r.status_code}")
+if r.status_code != 200:
+    print(r.text)
+
+response = json.loads(r.text)
+
+image_base64 = response[0]["image"]
+
+with open("image.png", "wb") as image_file:
+    image_file.write(base64.b64decode(image_base64))
+
+print("Output written to image.png")