| import os | |
| import logging | |
| import argparse | |
| import requests | |
| import numpy as np | |
| def send_image_post_request(ARGS: argparse.Namespace) -> None: | |
| logging.basicConfig(level=logging.INFO) | |
| list_test_images = os.listdir(ARGS.dir_test_images) | |
| for file_image in list_test_images: | |
| file_image = os.path.join(ARGS.dir_test_images, file_image) | |
| files = {"image_file": (file_image, open(file_image, "rb"))} | |
| # if the deployment is on local machine | |
| # response = requests.post( | |
| # "http://127.0.0.1:7860/predict", | |
| # files=files, | |
| # ) | |
| # if the deployment is on hugging face | |
| response = requests.post( | |
| "https://abhishekrs4-handwriting-recognition.hf.space/predict", | |
| files=files, | |
| ) | |
| logging.info(response.json()) | |
| return | |
| def main() -> None: | |
| dir_test_images = "./sample_test_images/" | |
| parser = argparse.ArgumentParser( | |
| formatter_class=argparse.ArgumentDefaultsHelpFormatter | |
| ) | |
| parser.add_argument( | |
| "--dir_test_images", | |
| default=dir_test_images, | |
| type=str, | |
| help="full directory path to dataset containing test images", | |
| ) | |
| ARGS, unparsed = parser.parse_known_args() | |
| send_image_post_request(ARGS) | |
| return | |
| if __name__ == "__main__": | |
| main() | |