diff --git a/examples/pneumonia_detection/infer.py b/examples/pneumonia_detection/infer.py new file mode 100644 index 0000000..b7c450e --- /dev/null +++ b/examples/pneumonia_detection/infer.py @@ -0,0 +1,8 @@ +from igel.medical.models.pneumonia_model import PneumoniaClassifier +import sys + +if __name__ == "__main__": + image_path = sys.argv[1] # pass image path as CLI arg + model = PneumoniaClassifier() + result = model.predict(image_path) + print(f"Prediction: {result}") diff --git a/igel/medical/models/config.yaml b/igel/medical/models/config.yaml new file mode 100644 index 0000000..e69de29 diff --git a/igel/medical/models/pneumonia_model.pth b/igel/medical/models/pneumonia_model.pth new file mode 100644 index 0000000..bdcd914 Binary files /dev/null and b/igel/medical/models/pneumonia_model.pth differ diff --git a/igel/medical/models/pneumonia_model.py b/igel/medical/models/pneumonia_model.py new file mode 100644 index 0000000..e69de29