File size: 442 Bytes
ab8b628 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 |
import sys
from image_classification_model.predict import predict
def main():
if len(sys.argv) < 2:
print("Usage: python predict.py <image_path>")
sys.exit(1)
image_path = sys.argv[1]
# Run prediction (handles preprocessing internally)
predicted_label = predict(image_path)
# Print output in Hugging Face-compatible format
print({"label": predicted_label})
if __name__ == "__main__":
main()
|