(model_id: str, image_file_name: str)
| 257 | return str(result) |
| 258 | @task |
| 259 | def image_classification(model_id: str, image_file_name: str) -> str: |
| 260 | inference = InferenceApi(repo_id=model_id, token=CONFIG["huggingface"]["token"]) |
| 261 | img_data = image_to_bytes(f"{DIRPATH}/{INPUT_PATH}/{image_file_name}") |
| 262 | result = inference(data=img_data) |
| 263 | return str(result) |
| 264 | @task |
| 265 | def image_to_text(model_id: str, image_file_name: str) -> str: |
| 266 | inference = InferenceApi(repo_id=model_id, token=CONFIG["huggingface"]["token"]) |
nothing calls this directly
no test coverage detected